Skip to content

Commit 31d826c

Browse files
authored
added correct path setting for the Books subset of MUSE. Previously, the News subset would (#112)
always be loaded.
1 parent 45c665b commit 31d826c

File tree

3 files changed

+10
-3
lines changed

3 files changed

+10
-3
lines changed

configs/experiment/unlearn/muse/default.yaml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,15 @@ data:
2424
args:
2525
hf_args:
2626
split: ${forget_split}
27+
path: muse-bench/MUSE-${data_split}
2728
retain:
2829
MUSE_retain:
2930
args:
3031
hf_args:
32+
path: muse-bench/MUSE-${data_split}
3133
split: ${retain_split}
3234

35+
3336
eval:
3437
muse:
3538
data_split: ${data_split}
@@ -48,4 +51,4 @@ trainer:
4851
# optim: paged_adamw_32bit
4952
# optim: adamw_torch
5053

51-
task_name: ???
54+
task_name: ???

configs/experiment/unlearn/muse/scalability.yaml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,13 @@ data:
2323
MUSE_forget_scal:
2424
args:
2525
hf_args:
26+
path: muse-bench/MUSE-${data_split}
2627
split: ${forget_split}
2728
retain:
2829
MUSE_retain:
2930
args:
3031
hf_args:
32+
path: muse-bench/MUSE-${data_split}
3133
split: ${retain_split}
3234

3335
eval:
@@ -48,4 +50,4 @@ trainer:
4850
# optim: paged_adamw_32bit
4951
# optim: adamw_torch
5052

51-
task_name: ???
53+
task_name: ???

configs/experiment/unlearn/muse/sustainabilty.yaml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,13 @@ data:
2323
MUSE_forget_sust:
2424
args:
2525
hf_args:
26+
path: muse-bench/MUSE-${data_split}
2627
split: ${forget_split}
2728
retain:
2829
MUSE_retain:
2930
args:
3031
hf_args:
32+
path: muse-bench/MUSE-${data_split}
3133
split: ${retain_split}
3234

3335
eval:
@@ -48,4 +50,4 @@ trainer:
4850
# optim: paged_adamw_32bit
4951
# optim: adamw_torch
5052

51-
task_name: ???
53+
task_name: ???

0 commit comments

Comments
 (0)