github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update train.sh 

改成真正能跑通的
pull/1288/head
曾小健 2023-06-30 11:02:17 +08:00 committed by GitHub
parent a9002a162a
commit 5d3f823bcc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ptuning/train.sh
View File

@ -1,7 +1,7 @@
PRE_SEQ_LEN=128 PRE_SEQ_LEN=128
LR=2e-2 LR=2e-2
CUDA_VISIBLE_DEVICES=0 python3 main.py \ CUDA_VISIBLE_DEVICES=0 torchrun main.py \
--do_train \ --do_train \
--train_file AdvertiseGen/train.json \ --train_file AdvertiseGen/train.json \
--validation_file AdvertiseGen/dev.json \ --validation_file AdvertiseGen/dev.json \