From 5d3f823bccd3b61e1858e9702cf7074f85886b04 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=BE=E5=B0=8F=E5=81=A5?= <2119516028@qq.com>
Date: Fri, 30 Jun 2023 11:02:17 +0800
Subject: [PATCH] Update train.sh
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

改成真正能跑通的
---
 ptuning/train.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ptuning/train.sh b/ptuning/train.sh
index efc9a16..bf75216 100644
--- 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 \