Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ChatGLM-6B_pytorch
Commits
cc3861a2
Commit
cc3861a2
authored
Aug 29, 2024
by
dcuai
Browse files
修改保存路径与eval对应
parent
49c37cd4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
ptuning/ft_train.sh
ptuning/ft_train.sh
+2
-2
No files found.
ptuning/ft_train.sh
View file @
cc3861a2
...
...
@@ -11,7 +11,7 @@ HIP_VISIBLE_DEVICES=0,1,2,3 deepspeed --num_gpus=4 --master_port $MASTER_PORT ma
--prompt_column
content
\
--response_column
summary
\
--model_name_or_path
../model/chatglm-6b
\
--output_dir
./output_ft/
nooptim-
adgen-chatglm-6b-ft-4c-
$LR
\
--output_dir
./output_ft/adgen-chatglm-6b-ft-4c-
$LR
\
--overwrite_output_dir
\
--max_source_length
512
\
--max_target_length
512
\
...
...
@@ -23,4 +23,4 @@ HIP_VISIBLE_DEVICES=0,1,2,3 deepspeed --num_gpus=4 --master_port $MASTER_PORT ma
--logging_steps
30
\
--save_steps
1000
\
--learning_rate
$LR
\
--fp16
\
\ No newline at end of file
--fp16
\
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment