p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: fix bug

This commit is contained in:
wql 2024-09-19 15:33:20 +08:00
parent 7d0cf5a79c
commit 618e57fa6b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
batch_run.sh
View File

@ -4,7 +4,7 @@
repeat_cnt=$1 repeat_cnt=$1
export ASCEND_RT_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 source /usr/local/Ascend/ascend-toolkit/set_env.sh
for ((i=0; i<repeat_cnt; i++)); do for ((i=0; i<repeat_cnt; i++)); do
bash run_once.sh lora_sft Qwen-7B 8 500 bash run_once.sh lora_sft Qwen-7B 8 500

2
run_once.sh
View File

@ -42,7 +42,7 @@ bash npu_status.sh ${output_dir} 60 0 &
npu_status_pid=$! npu_status_pid=$!
# echo "${npu_status_pid}" # echo "${npu_status_pid}"
if [ "${gpu_cnt}"="1" ]; then if [ "${gpu_cnt}" = "1" ]; then
ASCEND_RT_VISIBLE_DEVICES=0 llamafactory-cli train ${output_dir}/${run_name}.yaml | tee "${output_dir}/log.txt" & ASCEND_RT_VISIBLE_DEVICES=0 llamafactory-cli train ${output_dir}/${run_name}.yaml | tee "${output_dir}/log.txt" &
train_pid=$! train_pid=$!
echo "Start single npu train" echo "Start single npu train"