chore: modify tee

This commit is contained in:
wql 2024-09-24 09:41:24 +08:00
parent 87b6708a8b
commit 6d6f1cbdce
1 changed files with 2 additions and 2 deletions

View File

@ -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