Update train.sh

改成真正能跑通的
2023-06-30 11:02:17 +08:00 · 2023-06-30 11:02:17 +08:00 · 5d3f823bcc
parent a9002a162a
commit 5d3f823bcc
1 changed files with 1 additions and 1 deletions
--- a/ptuning/train.sh
+++ b/ptuning/train.sh
@ -1,7 +1,7 @@
 PRE_SEQ_LEN=128
 LR=2e-2

-CUDA_VISIBLE_DEVICES=0 python3 main.py \
+CUDA_VISIBLE_DEVICES=0 torchrun main.py \
    --do_train \
    --train_file AdvertiseGen/train.json \
    --validation_file AdvertiseGen/dev.json \