neural-amp-modeler

commit 288d69ce3948b8d0f35a861279ba13144d8eb5fe
parent 82bd6a65c5d8e094f62598d516fbe23413f9ac87
Author: Steven Atkinson <steven@atkinson.mn>
Date:   Mon, 23 Oct 2023 21:49:29 -0700

[Improvement] Report ESR instead of NRMSE in CLI trainer (#323)


Diffstat:
M
bin/train/main.py
 | 
4
+++-

1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/bin/train/main.py b/bin/train/main.py
@@ -100,7 +100,9 @@ def plot(
     # plt.plot(
     #     ds.y[window_start:window_end] - output[window_start:window_end], label="Error"
     # )
-    plt.title(f"NRMSE={_rms(torch.Tensor(output) - ds.y) / _rms(ds.y)}")
+    nrmse = _rms(torch.Tensor(output) - ds.y) / _rms(ds.y)
+    esr = nrmse**2
+    plt.title(f"ESR={esr:.3f}")
     plt.legend()
     if savefig is not None:
         plt.savefig(savefig)