From cc3fbff7043844f0826d0cd8e24e274c605d3cab Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Mon, 8 Jul 2019 19:11:53 -0400
Subject: [PATCH] moved sampler

---
 pytorch_lightning/models/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 39d0b761..08150120 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -176,6 +176,7 @@ 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