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
Deepseek_Math_7b_pytorch
Commits
a35dc7d8
Commit
a35dc7d8
authored
Jul 01, 2024
by
zhougaofeng
Browse files
Update multi_node.sh
parent
e5910c48
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
finetune/multi_node.sh
finetune/multi_node.sh
+1
-1
No files found.
finetune/multi_node.sh
View file @
a35dc7d8
...
...
@@ -16,7 +16,7 @@ HIP_VISIBLE_DEVICES=6,7 torchrun \
--master_port
$MASTER_PORT
\
src/train.py
\
--stage
sft
--do_train
\
--model_name_or_path
/home/practice/deepseek-math-7b-base
\
--model_name_or_path
/home/practice/deepseek-math-7b-base
\
<Your Path>
--dataset
alpaca_en_demo
--template
deepseek
--finetuning_type
lora
--lora_target
q_proj,v_proj
\
--output_dir
saves/deepseek/lora/sft
\
--overwrite_output_dir
\
...
...
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