neural-amp-modeler

commit 25f1639a8f45c4af7d7a2bb9748419782530afbe
parent 13b4d2363f6009222598bcaa26faacb307dcc133
Author: Steven Atkinson <steven@atkinson.mn>
Date:   Sun, 12 Feb 2023 16:18:20 -0800

Delay=0 in float interpolation, apply delay from slices

Diffstat:
M
nam/data.py
 | 
4
++++

1 file changed, 4 insertions(+), 0 deletions(-)
diff --git a/nam/data.py b/nam/data.py
@@ -138,6 +138,8 @@ def _interpolate_delay(
     """
     NOTE: This breaks the gradient tape!
     """
+    if delay == 0.0:
+        return x
     t_in = np.arange(len(x))
     n_out = len(x) - int(np.ceil(np.abs(delay)))
     if delay > 0:
@@ -361,6 +363,8 @@ class ParametricDataset(Dataset):
             c = deepcopy(config)
             start, stop, params = [s[k] for k in ("start", "stop", "params")]
             c.update(x=x[start:stop], y=y[start:stop], params=params)
+            if "delay" in s:
+                c["delay"] = s["delay"]
             datasets.append(ParametricDataset(**c))
         return ConcatDataset(datasets)