diff --git a/docs/LightningModule/methods.md b/docs/LightningModule/methods.md index 9163326e7f..d57c695034 100644 --- 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) ```