Update run-bench.py

YukeWang96 · Feb 25, 2021 · ee096e4 · ee096e4
1 parent c3d9754
commit ee096e4
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/run-bench.py b/run-bench.py
@@ -2,8 +2,6 @@
 
 import sys
 import os
-import subprocess
-
 
 models = [
     # 'VGG11',
@@ -18,20 +16,18 @@
 
 groups = [
     # 1,
-    # 2,
-    # 4,
-    8
+    2,
+    # 4
 ]
 
 overlaps = [
     # 0.25,
     0.50,
     # 0.75
 ]
+
 for model in models:
     for grp in groups:
         for oap in overlaps:
             print("=> {}, g: {}, o: {}".format(model, grp, oap))
             os.system("python main.py --model {} --groups {} --overlap {}".format(model, grp, oap))
-            # p = subprocess.Popen("python main.py --model {} --groups {} --overlap {}".format(model, grp, oap), shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate()[0].decode('utf-8')
-            # print("time: {}".format(p.split("\n")[-2].split(':')[-1]))