run_sentiment_cls.sh 333 字节
Newer Older
Z
Zeyu Chen 已提交
1
export CUDA_VISIBLE_DEVICES=5
2

3 4
CKPT_DIR="./ckpt_sentiment_cls"
python -u sentiment_cls.py \
5 6 7 8 9 10
                   --batch_size 32 \
                   --weight_decay  0.01 \
                   --checkpoint_dir $CKPT_DIR \
                   --num_epoch 3 \
                   --max_seq_len 128 \
                   --learning_rate 5e-5