Skip to content

Commit 8fbaccd

Browse files
added eval and train for redundancy (#464)
1 parent 32dd803 commit 8fbaccd

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

pytorch_lightning/root_module/root_module.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -247,6 +247,10 @@ def freeze(self):
247247
for param in self.parameters():
248248
param.requires_grad = False
249249

250+
self.eval()
251+
250252
def unfreeze(self):
251253
for param in self.parameters():
252254
param.requires_grad = True
255+
256+
self.train()

0 commit comments

Comments
 (0)