-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathgemma.sh
52 lines (49 loc) · 1.29 KB
/
gemma.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
python main.py \
data=stereoset \
model=gemma_last2 \
editor=malmen \
data.n_edits=64 \
data.batch_size=16 \
model_device=cuda:1 \
editor_device=cuda:0 \
early_stop_patience=4 \
editor.loc_coef=5.0
python main.py \
data=stereoset \
model=gemma_last2 \
editor=malmen \
data.n_edits=64 \
data.batch_size=16 \
model_device=cuda:1 \
editor_device=cuda:0 \
editor.load_checkpoint=True \
eval_only=True \
data.valid_path=dataset/stereoset/gender_test.json \
early_stop_patience=4 \
editor.loc_coef=5.0
python main.py \
data=stereoset \
model=gemma_last2 \
editor=malmen \
data.n_edits=64 \
data.batch_size=16 \
model_device=cuda:1 \
editor_device=cuda:0 \
editor.load_checkpoint=True \
eval_only=True \
data.valid_path=dataset/stereoset/race_test.json \
early_stop_patience=4 \
editor.loc_coef=5.0
python main.py \
data=stereoset \
model=gemma_last2 \
editor=malmen \
data.n_edits=64 \
data.batch_size=16 \
model_device=cuda:1 \
editor_device=cuda:0 \
editor.load_checkpoint=True \
eval_only=True \
data.valid_path=dataset/stereoset/religion_test.json \
early_stop_patience=4 \
editor.loc_coef=5.0