Skip to content

GitLab

Commit e9bec754 authored by zhaoying1's avatar zhaoying1
Browse files

Update run_train_single.sh

parent 9f69dd10
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ptuning/slurm_scripts/run_train_single.sh
View file @ e9bec754
...@@ -24,8 +24,8 @@ APP="python3 ../main.py \ ...@@ -24,8 +24,8 @@ APP="python3 ../main.py \
--deepspeed ../deepspeed.json \ --deepspeed ../deepspeed.json \
--do_train \ --do_train \
--train_file AdvertiseGen/train.json \ --train_file AdvertiseGen/train.json \
--prompt_column prompt \ --prompt_column content \
--response_column response \ --response_column summary \
--model_name_or_path THUDM/chatglm-6b \ --model_name_or_path THUDM/chatglm-6b \
--output_dir ./output_ft/pretrain \ --output_dir ./output_ft/pretrain \
--overwrite_output_dir \ --overwrite_output_dir \
......
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