From b644234d083844f82bdd495bc3808f6a7c2ba1a3 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 19:49:37 -0400 Subject: [PATCH] scaled batch size --- pytorch_lightning/models/trainer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index f98d70f770..5678efb0fa 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -192,8 +192,6 @@ 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: