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:
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