From 2018380598b2c0916d198260be236c714b74e894 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 7 Aug 2019 08:01:42 -0400
Subject: [PATCH] debug

---
 tests/test_models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_models.py b/tests/test_models.py
index 9142ca45..575ceb05 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -74,7 +74,7 @@ def test_cpu_restore_training():
     # we want to see if the weights come back correctly
     new_exp = get_exp(False, version=test_exp_version)
     trainer_options = dict(
-        max_nb_epochs=1,
+        max_nb_epochs=2,
         val_check_interval=0.50,
         val_percent_check=0.2,
         train_percent_check=0.2,