Commit e5b3db1f authored by zhaoying1's avatar zhaoying1
Browse files

Update pt_train.sh

parent 854699f0
......@@ -4,7 +4,7 @@ LR=5e-3
MASTER_PORT=$(shuf -n 1 -i 10000-65535)
CUDA_VISIBLE_DEVICES=0,1,2,3 deepspeed --master_port $MASTER_PORT main.py \
--deepspeed deepspeed_p.json \
--deepspeed deepspeed.json \
--do_train \
--train_file AdvertiseGen/train.json \
--test_file AdvertiseGen/dev.json \
......@@ -26,4 +26,4 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 deepspeed --master_port $MASTER_PORT main.py \
--learning_rate $LR \
--pre_seq_len $PRE_SEQ_LEN \
--fp16
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment