From 8e7d3c673776edf8f302019bd8685f7fc46ec03c Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Fri, 26 Jul 2019 23:16:03 -0400
Subject: [PATCH] added clean slurm save load test

---
 docs/LightningModule/methods.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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)
 ```