neural-amp-modeler

commit 86a264f0bfbddec73ac34807b1786bd9a2006c2f
parent c39a273e022b12c561b15049c3f29d76c78400b6
Author: Steven Atkinson <[email protected]>
Date:   Mon, 24 Apr 2023 19:03:57 -0700

Colab: Ensure model is on CPU before plotting prediction vs target (#209)

Model to CPU
Diffstat:
M
nam/train/core.py
 | 
1
+

1 file changed, 1 insertion(+), 0 deletions(-)
diff --git a/nam/train/core.py b/nam/train/core.py
@@ -421,6 +421,7 @@ def train(
             trainer.checkpoint_callback.best_model_path,
             **Model.parse_config(model_config),
         )
+    model.cpu()
     model.eval()
 
     _plot(