From 8e4fe2002b50a3cb49976b38ad85ff09da89e622 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 7 Aug 2019 07:45:57 -0400
Subject: [PATCH] fixed restore location

---
 tests/test_models.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/test_models.py b/tests/test_models.py
index 1708134aa6..6ee97ffe1a 100644
--- 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),
     )