#!/bin/bash

# bash batch_run.sh 3

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 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 lora_sft ChatGLM2-6B 8 500
    # bash run_once.sh lora_sft ChatGLM2-6B 8 1000
    # bash run_once.sh lora_sft ChatGLM2-6B 1 500
    # bash run_once.sh lora_sft ChatGLM2-6B 1 1000
    # bash run_once.sh inference ChatGLM2-6B 8 500
    # bash run_once.sh inference ChatGLM2-6B 8 1000
    # bash run_once.sh inference ChatGLM2-6B 1 500
    # bash run_once.sh inference ChatGLM2-6B 1 1000

    # bash run_once.sh lora_sft Llama2-7B 8 500
    # bash run_once.sh lora_sft Llama2-7B 8 1000
    # bash run_once.sh lora_sft Llama2-7B 1 500
    # bash run_once.sh lora_sft Llama2-7B 1 1000
    # bash run_once.sh inference Llama2-7B 8 500
    # bash run_once.sh inference Llama2-7B 8 1000
    # bash run_once.sh inference Llama2-7B 1 500
    # bash run_once.sh inference Llama2-7B 1 1000

done