[example] update gemini examples (#3868)

* [example]update gemini examples * [example]update gemini examples
2025-09-10 05:20:33 +00:00 · 2023-05-30 18:41:41 +08:00
parent 2506e275b8
commit 5f79008c4a
2 changed files with 29 additions and 30 deletions
--- a/examples/language/gpt/gemini/test_ci.sh
+++ b/examples/language/gpt/gemini/test_ci.sh
@@ -3,7 +3,7 @@ $(cd `dirname $0`;pwd)
 export TRAIN_STEP=4

 for MODEL_TYPE in "gpt2_medium"; do
-  for DISTPLAN in "colossalai"; do
+  for DISTPLAN in "CAI_Gemini"; do
    for BATCH_SIZE in 2; do
      for GPUNUM in 1 4; do
        for TPDEGREE in 1 2; do