From 5b59012123fed6ee18428ea893c1a02932586d08 Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Thu, 24 Oct 2024 13:40:15 +0800
Subject: [PATCH] chore: prepare batch run for 9g

---
 batch_run.sh | 25 +++++++++++++++++--------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/batch_run.sh b/batch_run.sh
index e74a46fe..aa154997 100644
--- a/batch_run.sh
+++ b/batch_run.sh
@@ -4,24 +4,33 @@
 
 repeat_cnt=$1
 
+export HCCL_CONNECT_TIMEOUT=300
+
 for ((i=0; i<repeat_cnt; i++)); do
+
+    bash run_once.sh lora_sft 9g-8B 8 500
+    bash run_once.sh lora_sft 9g-8B 8 1000
+    bash run_once.sh lora_sft 9g-8B 1 500
+    bash run_once.sh lora_sft 9g-8B 1 1000
+
+
     # 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 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 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 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 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
+    # 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
 
     # bash run_once.sh lora_sft ChatGLM2-6B 8 500
     # bash run_once.sh lora_sft ChatGLM2-6B 8 1000