ColossalAI/examples/language/gpt/gemini/test_ci.sh

set -x
$(cd `dirname $0`;pwd)
export TRAIN_STEP=4

for MODEL_TYPE in "gpt2_medium"; do
  for DISTPLAN in "CAI_Gemini"; do
    for BATCH_SIZE in 2; do
      for GPUNUM in 1 4; do
        MODEL_TYPE=${MODEL_TYPE} DISTPLAN=${DISTPLAN} BATCH_SIZE=${BATCH_SIZE} GPUNUM=${GPUNUM} \
        bash ./run_gemini.sh
      done
    done
  done

  for DISTPLAN in "CAI_ZeRO2" "CAI_ZeRO1"; do
    for BATCH_SIZE in 2; do
      for GPUNUM in 1 4; do
        MODEL_TYPE=${MODEL_TYPE} DISTPLAN=${DISTPLAN} BATCH_SIZE=${BATCH_SIZE} GPUNUM=${GPUNUM} \
        bash ./run_gemini.sh
        done
      done
    done
done