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

moved sampler

Browse Source
This commit is contained in:
William Falcon
2019-07-08 19:11:53 -04:00
parent 0bcc858cef
commit cc3fbff704
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
@@ -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