Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

moved sampler

This commit is contained in:
William Falcon 2019-07-08 19:15:28 -04:00
parent cc3fbff704
commit d596ff2039
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch_lightning/models/trainer.py
View File

@ -176,7 +176,6 @@ class Trainer(TrainerIO):
# determine number of training batches
self.nb_tng_batches = len(self.tng_dataloader)
print('tng batches: ', self.nb_tng_batches, self.nb_test_batches, self.nb_val_batches)
self.nb_tng_batches = int(self.nb_tng_batches * self.train_percent_check)
# determine number of validation batches
@ -192,6 +191,8 @@ class Trainer(TrainerIO):
# determine when to check validation
self.val_check_batch = int(self.nb_tng_batches * self.val_check_interval)
print('tng batches: ', self.nb_tng_batches, self.nb_test_batches, self.nb_val_batches)
def __add_tqdm_metrics(self, metrics):
for k, v in metrics.items():
if type(v) is torch.Tensor: