added clean slurm save load test

2026-09-09 11:32:07 +08:00 · 2019-07-26 23:16:03 -04:00
parent 4cacb5a21b
commit 8e7d3c6737
1 changed files with 2 additions and 1 deletions
@@ -21,7 +21,8 @@ pretrained_model = MyLightningModule.load_from_metrics(
    map_location=None
 )
    
-# predict 
+# predict
+pretrained_model.eval()
 pretrained_model.freeze()
 y_hat = pretrained_model(x)
 ```