From 0365da13d2caf7f5649de148d339dfe131375477 Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Wed, 23 Oct 2024 20:35:24 +0800
Subject: [PATCH] chore: move code

---
 batch_run.sh | 2 --
 run_once.sh  | 4 +++-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/batch_run.sh b/batch_run.sh
index 2583073d..e74a46fe 100644
--- a/batch_run.sh
+++ b/batch_run.sh
@@ -4,8 +4,6 @@
 
 repeat_cnt=$1
 
-source /usr/local/Ascend/ascend-toolkit/set_env.sh
-
 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 1000
diff --git a/run_once.sh b/run_once.sh
index d4413f0d..10817108 100644
--- a/run_once.sh
+++ b/run_once.sh
@@ -3,7 +3,7 @@
 
 # bash run_once.sh lora_sft Qwen-7B 8 50
 # bash run_once.sh lora_sft Qwen-7B 1 50
-# bash run_once.sh lora_sft 9g-8B 8 50
+# bash run_once.sh lora_sft 9g-8B 8 500
 # bash run_once.sh inference Qwen-7B 8 50
 run_type="$1"
 model="$2"
@@ -17,6 +17,8 @@ echo "max_steps: ${max_steps}"
 
 current_datetime=$(date +%Y%m%d%H%M%S)
 
+source /usr/local/Ascend/ascend-toolkit/set_env.sh
+
 if [ "${run_type}"="lora_sft" ]; then
     run_name="${run_type}_${model}_${gpu_cnt}_gpu_${max_steps}_step_${current_datetime}"