-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathxnli.sh
24 lines (24 loc) · 1.44 KB
/
xnli.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
#!/bin/bash
DEVICE=0
SEED=42
CUDA_VISIBLE_DEVICES=$DEVICE python run_xnli.py --output_dir xnli-result/FT/32shots/1 \
--seed $SEED \
--max_train_samples 32 \
--max_eval_samples 32 \
--train_language en \
--overwrite_output_dir \
--do_train \
--do_predict \
--language ar,bg,de,el,en,es,fr,hi,ru,sw,th,tr,ur,vi,zh \
--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 8 \
--per_device_eval_batch_size 32 \
--gradient_accumulation_steps 1 \
--learning_rate 1e-5 \
--lr_scheduler_type constant \
--num_train_epochs 50