removed from_lightning flag

This commit is contained in:
William Falcon
2019-07-08 20:14:56 -04:00
parent f38f3827fd
commit 611fbdea3e
2 changed files with 0 additions and 2 deletions
-1
View File
@@ -214,7 +214,6 @@ 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)
@@ -21,7 +21,6 @@ class LightningModule(GradInformation, ModelIO, OptimizerConfig, ModelHooks):
self.global_step = 0
self.loaded_optimizer_states_dict = {}
self.trainer = None
self.from_lightning = True
self.experiment = None
# track if gpu was requested for checkpointing