Skip to content

GitLab

Commit c6186d00 authored by zhaoying1's avatar zhaoying1
Browse files

Update ptuning_train.sh

parent 0d7ae9df
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ptuning/ptuning_train.sh
View file @ c6186d00
...@@ -3,7 +3,7 @@ LR=5e-3 ...@@ -3,7 +3,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 \ HIP_VISIBLE_DEVICES=0,1,2,3 deepspeed --master_port $MASTER_PORT main.py \
--deepspeed deepspeed.json \ --deepspeed deepspeed.json \
--do_train \ --do_train \
--train_file AdvertiseGen/train.json \ --train_file AdvertiseGen/train.json \
......
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