Merge pull request #1 from claudiofahey/dsws

Minor update to testgen.py for single workstation
claudiofahey · Mar 20, 2020 · ca7f5d2 · ca7f5d2
2 parents 5c022b9 + cb0eb6e
commit ca7f5d2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/testgen.py b/testgen.py
@@ -117,8 +117,8 @@ def add_test():
 # Full test suite
 for repeat in range(3):
     for storage_type in ['isilon']:     # 'isilon','filestore'
-        for cached in [True]:
-            for model in ['resnet50']:  # 'vgg16','resnet152','inception3','inception4'
+        for cached in [True] if storage_type=='local' else [False,True]:
+            for model in ['resnet50','vgg16','resnet152','inception3','inception4']:
                 for batch_group_size in [10]:
                     for batch_size in [64]:
                         for data_dir_template_count in [1 if cached or storage_type=='filestore' else 4]: