-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsh_run_av.sh
29 lines (29 loc) · 849 Bytes
/
sh_run_av.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
#Intensive module
export SQUAD_DIR=data/uit-visquad
export TRAIN_FILE=train-3.json
export DEV_FILE=dev-3.json
export PREDICT_FILE=ptest.json
python ./run_av.py \
--model_type xlm-roberta-dep2 \
--model_name_or_path xlm-roberta-large \
--do_train \
--do_eval \
--do_lower_case \
--version_2_with_negative \
--train_file $SQUAD_DIR/$TRAIN_FILE \
--predict_file $SQUAD_DIR/$DEV_FILE \
--learning_rate 2e-5 \
--num_train_epochs 10 \
--max_seq_length 384 \
--doc_stride 128 \
--max_query_length=64 \
--per_gpu_train_batch_size=8 \
--per_gpu_eval_batch_size=16 \
--warmup_steps=814 \
--output_dir models/uit-visquad/xlm-r-dep2-3 \
--eval_all_checkpoints \
--save_steps 5000 \
--n_best_size=20 \
--max_answer_length=50 \
--overwrite_output_dir \
--overwrite_cache