Skip to content

GitLab

Commit 0ba927e4 authored by wanglch's avatar wanglch
Browse files

Update multi_dcu_train.sh

parent b81b5953
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
multi_dcu_train.sh
View file @ 0ba927e4
#!/bin/bash #!/bin/bash
## 如果需要指定显卡训练可将 --num_gpus 4 改为 --include="localhost:0,3" ## 如果需要指定显卡训练可将 --num_gpus 4 改为 --include="localhost:0,3"
deepspeed --master_port 25678 --num_gpus 4 LLaMA-Factory-main/src/train_bash.py \ deepspeed --master_port 25678 --num_gpus 4 LLaMA-Factory-main/src/train_bash.py \
--deepspeed /LLaMA-Factory-main/deepspeed.json \ --deepspeed LLaMA-Factory-main/deepspeed.json \
--stage sft \ --stage sft \
--do_train \ --do_train \
--model_name_or_path THUDM/chatglm2-6b \ --model_name_or_path THUDM/chatglm2-6b \
......
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