Adjust parameters for GCP testing

Disabled NVLink. Updated Prometheus for single workstation.
claudiofahey · Mar 20, 2020 · cb0eb6e · cb0eb6e
1 parent 20bc1fb
commit cb0eb6e
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]: