run_sequence_label.sh 462 字节
Newer Older
1
export FLAGS_eager_delete_tensor_gb=0.0
K
kinghuin 已提交
2
export CUDA_VISIBLE_DEVICES=0
3

Z
Zeyu Chen 已提交
4
CKPT_DIR="./ckpt_sequence_label"
Z
Zeyu Chen 已提交
5
python -u sequence_label.py \
6 7 8
                   --batch_size 16 \
                   --weight_decay  0.01 \
                   --num_epoch 3 \
Z
Zeyu Chen 已提交
9
                   --checkpoint_dir $CKPT_DIR \
10
                   --max_seq_len 128 \
W
wuzewu 已提交
11 12 13
                   --learning_rate 5e-5 \
                   --use_pyreader True \
                   --use_data_parallel True