neural-amp-modeler

commit 5c727e07c4c1745f713995a2a501ff53a7cd986e
parent 8191f36c97261cb6e24967e45332aab2c638adcd
Author: Steven Atkinson <steven@atkinson.mn>
Date:   Sat,  4 Mar 2023 11:31:21 -0600

Colab exported model indices start at zero (#113)


Diffstat:
M
nam/train/colab.py
 | 
8
++++----

1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/nam/train/colab.py b/nam/train/colab.py
@@ -53,7 +53,7 @@ def _get_valid_export_directory():
     def get_path(version):
         return Path("exported_models", f"version_{version}")
 
-    version = 1
+    version = 0
     while get_path(version).exists():
         version += 1
     return get_path(version)
@@ -63,9 +63,9 @@ def run(
     epochs: int = 100,
     delay: Optional[int] = None,
     architecture: str = "standard",
-    lr: float=0.004,
-    lr_decay: float=0.007,
-    seed: Optional[int]=0,
+    lr: float = 0.004,
+    lr_decay: float = 0.007,
+    seed: Optional[int] = 0,
 ):
     """
     :param epochs: How amny epochs we'll train for.