wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-27 13:12:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

removed validation call

Browse Source
This commit is contained in:
William Falcon committed 2019-07-15 14:56:56 -04:00
1 parent 6a33f0d483
commit 58e6199ce8
1 file changed
-3
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
-3
View File
@@ -216,9 +216,6 @@ class Trainer(TrainerIO):
:param max_batches: Scalar
:return:
"""
if self.proc_rank == 0:
print('validating...')
# enable eval mode
model.zero_grad()
model.eval()