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
744a3911
Commit
744a3911
authored
Jul 01, 2024
by
zhougaofeng
Browse files
Update single_node.sh
parent
a35dc7d8
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
finetune/single_node.sh
finetune/single_node.sh
+1
-1
No files found.
finetune/single_node.sh
View file @
744a3911
...
...
@@ -3,7 +3,7 @@
export
HIP_VISIBLE_DEVICES
=
6
python 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_cache
--per_device_train_batch_size
2
--gradient_accumulation_steps
32
--lr_scheduler_type
cosine
\
...
...
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