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

Error if dataset size = 1 batch. (#141)

Browse Source 
Fix for the bug mentioned in https://github.com/williamFalcon/pytorch-lightning/issues/139
This commit is contained in:
sebftw
2019-08-18 18:15:58 -04:00
committed by William Falcon
parent e646d745da
commit 26d3f0dbea
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+1
View File
@@ -370,6 +370,7 @@ class Trainer(TrainerIO):
# determine when to check validation
self.val_check_batch = int(self.nb_tng_batches * self.val_check_interval)
self.val_check_batch = max(1, self.val_check_batch)
def __add_tqdm_metrics(self, metrics):
for k, v in metrics.items():