From e89f1022afe42921f8a1ab3654ce2a6713163135 Mon Sep 17 00:00:00 2001
From: heavyrain_lzy <1528794076@qq.com>
Date: Fri, 28 Jun 2024 10:46:09 +0800
Subject: [PATCH] change max_steps

---
 .../llama2/pretrain_config_llama2_7b/pretrain-llama2_7b.json    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_tipc/static/auto_parallel/llama2/pretrain_config_llama2_7b/pretrain-llama2_7b.json b/tests/test_tipc/static/auto_parallel/llama2/pretrain_config_llama2_7b/pretrain-llama2_7b.json
index e9d517b44a2f..ae1e3012274d 100644
--- a/tests/test_tipc/static/auto_parallel/llama2/pretrain_config_llama2_7b/pretrain-llama2_7b.json
+++ b/tests/test_tipc/static/auto_parallel/llama2/pretrain_config_llama2_7b/pretrain-llama2_7b.json
@@ -26,7 +26,7 @@
   "min_learning_rate": 3e-06,
   "warmup_steps": 30,
   "logging_steps": 10,
-  "max_steps": 50,
+  "max_steps": 500,
   "save_steps": 5000,
   "eval_steps": 1000,
   "weight_decay": 0.01,