From 8fbaccddaed6dc574e02fbe5d2f3dae161336785 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 5 Nov 2019 09:14:33 -0500 Subject: [PATCH] added eval and train for redundancy (#464) --- pytorch_lightning/root_module/root_module.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pytorch_lightning/root_module/root_module.py b/pytorch_lightning/root_module/root_module.py index b25ab9e4..fc8d62c5 100644 --- a/pytorch_lightning/root_module/root_module.py +++ b/pytorch_lightning/root_module/root_module.py @@ -247,6 +247,10 @@ class LightningModule(GradInformation, ModelIO, ModelHooks): for param in self.parameters(): param.requires_grad = False + self.eval() + def unfreeze(self): for param in self.parameters(): param.requires_grad = True + + self.train()