neural-amp-modeler

commit 44bb9e656148586d6a210b57bea59343baf30212
parent 72916e5ae5bb0f0cd3a02ad3165648ac5daa52ea
Author: Steven Atkinson <steven@atkinson.mn>
Date:   Sun, 24 Apr 2022 10:36:41 -0700

Y scale, decrease batch size (trains better)

Diffstat:
M
bin/train/inputs/config_learning.json
 | 
2
+-
M
nam/data.py
 | 
2
++

2 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/bin/train/inputs/config_learning.json b/bin/train/inputs/config_learning.json
@@ -1,6 +1,6 @@
 {
     "train_dataloader": {
-        "batch_size": 32,
+        "batch_size": 16,
         "shuffle": true,
         "pin_memory": true,
         "drop_last": true,
diff --git a/nam/data.py b/nam/data.py
@@ -121,6 +121,7 @@ class Dataset(AbstractDataset, InitializableFromConfig):
         start: Optional[int] = None,
         stop: Optional[int] = None,
         delay: Optional[int] = None,
+        y_scale: float = 1.0,
     ):
         """
         :param start: In samples
@@ -135,6 +136,7 @@ class Dataset(AbstractDataset, InitializableFromConfig):
             else:
                 x = x[-delay:]
                 y = y[:delay]
+        y = y * y_scale
         self._validate_inputs(x, y, nx, ny)
         self._x = x
         self._y = y