Fix the number of training batches used in the training loop (#653)

* Fix the number of processed training batches

* Fix tests

* fix tests

* fix tests

* One more attempt

* Fix another test
This commit is contained in:
Vadim Bereznyuk
2020-01-05 14:37:09 -05:00
committed by William Falcon
parent 7824b5c5f5
commit 12edc3099c
4 changed files with 9 additions and 10 deletions
+1 -1
View File
@@ -309,7 +309,7 @@ def test_custom_logger(tmpdir):
trainer_options = dict(
max_epochs=1,
train_percent_check=0.01,
train_percent_check=0.05,
logger=logger,
default_save_path=tmpdir
)