added clean slurm save load test

2019-07-26 23:16:03 -04:00 · 2019-07-26 23:16:03 -04:00 · 8e7d3c6737
parent 4cacb5a21b
commit 8e7d3c6737
1 changed files with 2 additions and 1 deletions
--- a/docs/LightningModule/methods.md
+++ b/docs/LightningModule/methods.md
@ -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)
 ```