Skip to content

GitLab

Commit 6724afb6 authored by zhaoying1's avatar zhaoying1
Browse files

Update ptuning_train.sh

parent 3392f187
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
ptuning/ptuning_train.sh
View file @ 6724afb6
...@@ -10,7 +10,6 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 deepspeed --master_port $MASTER_PORT main.py \ ...@@ -10,7 +10,6 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 deepspeed --master_port $MASTER_PORT main.py \
--test_file AdvertiseGen/dev.json \ --test_file AdvertiseGen/dev.json \
--prompt_column content \ --prompt_column content \
--response_column summary \ --response_column summary \
--overwrite_cache \
--model_name_or_path THUDM/chatglm-6b \ --model_name_or_path THUDM/chatglm-6b \
--output_dir ./output_pt/adgen-chatglm-6b-pt-4c-$LR \ --output_dir ./output_pt/adgen-chatglm-6b-pt-4c-$LR \
--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