Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added clean slurm save load test

This commit is contained in:
William Falcon 2019-07-26 23:16:03 -04:00
parent 4cacb5a21b
commit 8e7d3c6737
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docs/LightningModule/methods.md
View File

@ -21,7 +21,8 @@ pretrained_model = MyLightningModule.load_from_metrics(
map_location=None map_location=None
) )
# predict # predict
pretrained_model.eval()
pretrained_model.freeze() pretrained_model.freeze()
y_hat = pretrained_model(x) y_hat = pretrained_model(x)
``` ```