smart ratio v3 init

args_helper.py

@@ -904,6 +904,37 @@ def parse_arguments(self, jupyter_mode=False):
             type=int,
             help="smart ratio version number (1, 2, ...)",
         )
+        parser.add_argument(
+            "--srV3-epoch",
+            default=160,
+            type=int,
+            help="number of intermediate epochs used for loading srV3",
+        )
+        parser.add_argument(
+            "--sr_seq",
+            default=None,
+            type=str,
+            help="sequence to test for smart ratio",
+        )
+
+        '''
+        parser.add_argument(
+            "--current-layer",
+            default=0,
+            type=int,
+            help="current layer index",
+        )
+        '''
+        # parser.add_argument(
+        #     "--pt-sr-init",
+        #     type=list,
+        #     default=[],
+        #     nargs="+",
+        #     metavar="M",
+        #     help="List of initial sparsity pattern for pt-sr (resnet20)"
+        # )
+
+
 
         if jupyter_mode:
             args = parser.parse_args("")

cifar_exec.sh

@@ -64,21 +64,21 @@ BLOCK
 
 # Running trials in parallel
 # NOTE: make sure to delete/comment subfolder from the config file or else it may not work
-conf_file="configs/hypercube/wideresnet28/wideresnet28_weight_training.yml"
-log_root="wideresnet28_"
+conf_file="configs/sr/resnet20/resnet20_find_srV3.yml"
+log_root="pt_sr_debug_"
 log_end="_log"
-subfolder_root="wideresnet28_results_trial_"
+subfolder_root="pt_sr_debug_"
 
-for trial in 2 3
+for trial in 1
 do
     python main.py \
     --config "$conf_file" \
     --trial-num $trial \
-    --subfolder "$subfolder_root$trial" > "$log_root$trial$log_end" 2>&1 &
-
-    python main.py \
-    --config "$conf_file" \
-    --trial-num $trial \
-    --invert-sanity-check \
-    --subfolder "invert_$subfolder_root$trial" > "invert_$log_root$trial$log_end" 2>&1 &
-done
+    --subfolder "$subfolder_root$trial" > "$log_root$trial$log_end" 2>&1
+
+    #    python main.py \
+    #    --config "$conf_file" \
+    #    --trial-num $trial \
+    #    --invert-sanity-check \
+    #    --subfolder "invert_$subfolder_root$trial" > "invert_$log_root$trial$log_end" 2>&1 &
+done