fixed restore location

2019-08-07 07:45:57 -04:00 · 2019-08-07 07:45:57 -04:00 · 8e4fe2002b
parent b0fae55571
commit 8e4fe2002b
1 changed files with 3 additions and 0 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -75,6 +75,9 @@ def test_cpu_restore_training():
    new_exp = get_exp(False, version=test_exp_version)
    trainer_options = dict(
        max_nb_epochs=1,
+        val_check_interval=0.50,
+        val_percent_check=0.2,
+        train_percent_check=0.2,
        experiment=new_exp,
        checkpoint_callback=ModelCheckpoint(save_dir),
    )