neural-amp-modeler

commit 5f24710e029eee802b85c0584e49c9927f60a7a8
parent e8d3dbede3c532b7b079a9c948abe8a7a3f8664f
Author: Steven Atkinson <steven@atkinson.mn>
Date:   Sun, 24 Apr 2022 21:36:45 -0700

Fix bug when no checkpoints

Diffstat:
M
bin/train/main.py
 | 
9
++++++---

1 file changed, 6 insertions(+), 3 deletions(-)
diff --git a/bin/train/main.py b/bin/train/main.py
@@ -118,9 +118,12 @@ def main(args):
         **learning_config.get("trainer_fit_kwargs", {}),
     )
     # Go to best checkpoint
-    model = Model.load_from_checkpoint(
-        trainer.checkpoint_callback.best_model_path, **Model.parse_config(model_config)
-    )
+    best_checkpoint = trainer.checkpoint_callback.best_model_path
+    if best_checkpoint != "":
+        model = Model.load_from_checkpoint(
+            trainer.checkpoint_callback.best_model_path,
+            **Model.parse_config(model_config),
+        )
     model.eval()
     plot(
         model,