Skip to content

GitLab

Commit e5b3db1f authored by zhaoying1's avatar zhaoying1
Browse files

Update pt_train.sh

parent 854699f0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ptuning/pt_train.sh
View file @ e5b3db1f
...@@ -4,7 +4,7 @@ LR=5e-3 ...@@ -4,7 +4,7 @@ LR=5e-3
MASTER_PORT=$(shuf -n 1 -i 10000-65535) MASTER_PORT=$(shuf -n 1 -i 10000-65535)
CUDA_VISIBLE_DEVICES=0,1,2,3 deepspeed --master_port $MASTER_PORT main.py \ CUDA_VISIBLE_DEVICES=0,1,2,3 deepspeed --master_port $MASTER_PORT main.py \
--deepspeed deepspeed_p.json \ --deepspeed deepspeed.json \
--do_train \ --do_train \
--train_file AdvertiseGen/train.json \ --train_file AdvertiseGen/train.json \
--test_file AdvertiseGen/dev.json \ --test_file AdvertiseGen/dev.json \
...@@ -26,4 +26,4 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 deepspeed --master_port $MASTER_PORT main.py \ ...@@ -26,4 +26,4 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 deepspeed --master_port $MASTER_PORT main.py \
--learning_rate $LR \ --learning_rate $LR \
--pre_seq_len $PRE_SEQ_LEN \ --pre_seq_len $PRE_SEQ_LEN \
--fp16 --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