From d596ff2039753457096ed74137d3f391bf2ef14f Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 19:15:28 -0400 Subject: [PATCH] moved sampler --- pytorch_lightning/models/trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 081501206b..ad90080de2 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -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: