add ablation study

khairulislam · Dec 3, 2024 · fb55213 · fb55213
1 parent 29caecd
commit fb55213
Show file tree

Hide file tree

Showing 79 changed files with 236,737 additions and 1 deletion.
diff --git a/interpret.py b/interpret.py
@@ -30,7 +30,7 @@ def main(args):
     for itr_no in range(1, args.itrs+1):
         if (original_itr is not None) and original_itr != itr_no: continue
 
-        args.seed = experiment_seeds[itr_no-1]
+        args.seed = int(experiment_seeds[itr_no-1])
         print(f'\n>>>> itr_no: {itr_no}, seed: {args.seed} <<<<<<')
         set_random_seed(args.seed)
         args.itr_no = itr_no

diff --git a/requirements.txt b/requirements.txt
@@ -10,4 +10,6 @@ time-interpret # ==0.3.0
 captum
 tqdm
 einops
+peft
+transformers
 # install pytorch and cuda following readme
-Original file line number
+Diff line change
@@ Expand Up / @@ -10,4 +10,6 @@ time-interpret # ==0.3.0 @@
     captum
     tqdm
     einops
+    peft
+    transformers
     # install pytorch and cuda following readme