From e801914d1dd7ef625dd7f8e22bb0bd5dfae230ff Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Tue, 25 Jun 2019 19:18:27 -0400
Subject: [PATCH] updated args

---
 pytorch_lightning/models/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index c8d23cea..91ed48c6 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -163,6 +163,7 @@ class Trainer(TrainerIO):
         # enable eval mode
         model.zero_grad()
         model.eval()
+        model.from_lightning = True
 
         # disable gradients to save memory
         torch.set_grad_enabled(False)