neural-amp-modeler

Neural network emulator for guitar amplifiers
Log | Files | Refs | README | LICENSE

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:
Mbin/train/inputs/config_learning.json | 2+-
Mnam/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