From 6d6f1cbdce856eda2498c4b3ca43b09c2e128fb5 Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Tue, 24 Sep 2024 09:41:24 +0800
Subject: [PATCH] chore: modify tee

---
 run_once.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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