Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
root committed Dec 27, 2024
1 parent 0607554 commit 0bb5113
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -115,12 +115,12 @@ function _train(){
"
fi
case ${run_process_type} in
SingleP) train_cmd="accelerate launch --config_file n1c1.yaml ${train_cmd}" ;;
SingleP) train_cmd="accelerate launch ${train_cmd}" ;;
MultiP)
if [ ${device_num:3} = '32' ];then
train_cmd="accelerate launch --config_file n4c32.yaml --num_processes ${num_workers} --num_machines ${node_num} --machine_rank ${node_rank} --main_process_ip ${master_addr} --main_process_port ${master_port} ${train_cmd}"
train_cmd="accelerate launch --num_processes ${num_workers} --num_machines ${node_num} --machine_rank ${node_rank} --main_process_ip ${master_addr} --main_process_port ${master_port} ${train_cmd}"
else
train_cmd="accelerate launch --config_file n1c8.yaml ${train_cmd}"
train_cmd="accelerate launch ${train_cmd}"
fi;;
*) echo "choose run_mode(SingleP or MultiP)"; exit 1;
esac
Expand Down

0 comments on commit 0bb5113

Please sign in to comment.