-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdpa-pawsx.sh
29 lines (29 loc) · 1.66 KB
/
dpa-pawsx.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
#!/bin/bash
DEVICE=0
SEED=42
((NUM=SEED-41))
CUDA_VISIBLE_DEVICES=$DEVICE python run-prompt.py --output_dir pa-result/mixup-alpha1.2/1 \
--dataset paws-x \
--seed $SEED \
--tune_LM \
--mixup_strategy none \
--mixup_alpha 1.2 \
--multi_lingual_optim \
--max_train_samples 128 \
--max_eval_samples 128 \
--train_language en \
--do_train \
--do_predict \
--language de,en,es,fr,zh,ja,ko \
--save_strategy epoch \
--evaluation_strategy epoch \
--load_best_model_at_end \
--metric_for_best_model eval_accuracy \
--greater_is_better True \
--save_total_limit 2 \
--per_device_train_batch_size 2 \
--per_device_eval_batch_size 8 \
--gradient_accumulation_steps 4 \
--learning_rate 1e-5 \
--lr_scheduler_type constant \
--num_train_epochs 50