Merge pull request #77 from undertherain/testing

Add missing --nb_epoch=1 to some tests
undertherain · Jul 6, 2020 · ba404ec · ba404ec
2 parents 9898f48 + 47e6b5f
commit ba404ec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/test/tensorflow/test_vgg19.py b/test/tensorflow/test_vgg19.py
@@ -14,6 +14,7 @@ def setUp(self):
             "--framework=tensorflow",
             "--problem_size=4",
             "--batch_size=2",
+            "--nb_epoch=1",
         ]
 
     def test_vgg19(self):

diff --git a/test/tensorflow/test_xception.py b/test/tensorflow/test_xception.py
@@ -14,6 +14,7 @@ def setUp(self):
             "--framework=tensorflow",
             "--problem_size=4",
             "--batch_size=2",
+            "--nb_epoch=1",
         ]
 
     def test_xception(self):