ready

Tune.py

@@ -232,7 +232,8 @@ def run_unlearning_and_strategy_eval(env_dict, forget_class_idx, strategy, evalu
             forget_test_loader=forget_test_loader,
             unlearned_instance=reloaded,            
             base_shadow_instance=shadow_model,  
-            device=device
+            device=device,
+            strategy_in_use = strategy_in_use
         )
     
 
@@ -248,7 +249,7 @@ def run_unlearning_and_strategy_eval(env_dict, forget_class_idx, strategy, evalu
 # entry
 if __name__ == "__main__":
 
-    outer_loop = 10
+    outer_loop = 20
 
     inner_loop = CLASS_SIZE
     
@@ -307,10 +308,10 @@ if __name__ == "__main__":
             )
 
             strategies = [
-                retrain,
+                #retrain,
                 linear_filtration,
-                weight_filtration,
-                certified_unlearning,
+                #weight_filtration,
+                #certified_unlearning,
             ]
             suite_runner = UnlearningAttack(arch=ARCH, class_size=CLASS_SIZE)
                 # Unlearning Iteration