diff --git a/run_once.sh b/run_once.sh index 0f740e41..960a8a59 100644 --- a/run_once.sh +++ b/run_once.sh @@ -43,11 +43,11 @@ npu_status_pid=$! # echo "${npu_status_pid}" 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 2>&1 | tee -i "${output_dir}/log.txt" & train_pid=$! echo "Start single npu train" else - FORCE_TORCHRUN=1 llamafactory-cli train ${output_dir}/${run_name}.yaml | tee "${output_dir}/log.txt" & + FORCE_TORCHRUN=1 llamafactory-cli train ${output_dir}/${run_name}.yaml 2>&1 | tee -i "${output_dir}/log.txt" & train_pid=$! echo "Start multi npu train" fi