From e99fab5e4ce2da6902f6a24d71ed04fde4770921 Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Thu, 19 Sep 2024 14:03:17 +0800
Subject: [PATCH] chore: modify batch run

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

diff --git a/batch_run.sh b/batch_run.sh
index fd9d5f59..56c43e92 100644
--- a/batch_run.sh
+++ b/batch_run.sh
@@ -1 +1,25 @@
-bash run_once.sh lora_sft Qwen-7B 4 50
+#!/bin/bash
+
+# bash batch_run.sh 3
+
+repeat_cnt=$1
+
+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
+    bash run_once.sh lora_sft Qwen-7B 1 500
+    bash run_once.sh lora_sft Qwen-7B 1 1000
+    bash run_once.sh lora_sft Baichuan2-7B 8 500
+    bash run_once.sh lora_sft Baichuan2-7B 8 1000
+    bash run_once.sh lora_sft Baichuan2-7B 1 500
+    bash run_once.sh lora_sft Baichuan2-7B 1 1000
+
+    bash run_once.sh inference Qwen-7B 8 500
+    bash run_once.sh inference Qwen-7B 8 1000
+    bash run_once.sh inference Qwen-7B 1 500
+    bash run_once.sh inference Qwen-7B 1 1000
+    bash run_once.sh inference Baichuan2-7B 8 500
+    bash run_once.sh inference Baichuan2-7B 8 1000
+    bash run_once.sh inference Baichuan2-7B 1 500
+    bash run_once.sh inference Baichuan2-7B 1 1000
+done
diff --git a/run_once.sh b/run_once.sh
index e92e0705..e859f6e5 100644
--- a/run_once.sh
+++ b/run_once.sh
@@ -50,6 +50,6 @@ fi
 wait $train_pid
 echo "Train ended"
 
-# sleep 60
+sleep 60
 kill $npu_status_pid
 echo "Npu status ended"
\ No newline at end of file