wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-27 13:12:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

added clean slurm save load test

Browse Source
This commit is contained in:
William Falcon committed 2019-07-26 23:16:03 -04:00
1 parent 4cacb5a21b
commit 8e7d3c6737
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
docs/LightningModule/methods.md
+2 -1
View File
@@ -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)
```