run_thuc.sh 1.2 KB
Newer Older
T
tianxin04 已提交
1 2 3 4 5 6 7 8 9 10
set -eux

export FLAGS_sync_nccl_allreduce=1
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7

python -u run_classifier.py \
                   --use_cuda true \
                   --do_train true \
                   --do_val true \
                   --do_test true \
T
tianxin 已提交
11
                   --verbose true \
T
tianxin04 已提交
12
                   --batch_size 8 \
T
tianxin04 已提交
13
                   --init_pretraining_params ${MODEL_PATH}/params \
T
tianxin 已提交
14 15
                   --train_set ${TASK_DATA_PATH}/thuc/train.tsv \
                   --dev_set ${TASK_DATA_PATH}/thuc/dev.tsv,${TASK_DATA_PATH}/thuc/test.tsv \
T
tianxin04 已提交
16 17
                   --vocab_path config/vocab.txt \
                   --ernie_config_path config/ernie_config.json \
T
tianxin 已提交
18
                   --checkpoints ./checkpoints \
T
tianxin04 已提交
19 20 21
                   --save_steps 1000 \
                   --weight_decay  0.01 \
                   --warmup_proportion 0.0 \
T
tianxin 已提交
22
                   --validation_steps 100 \
T
tianxin04 已提交
23 24
                   --epoch 3 \
                   --max_seq_len 512 \
T
tianxin 已提交
25
                   --learning_rate 1.5e-5 \
T
tianxin04 已提交
26 27
                   --skip_steps 10 \
                   --num_iteration_per_drop_scope 1 \
T
tianxin 已提交
28
                   --num_labels 10 \
T
tianxin04 已提交
29
                   --random_seed 1