From 815678f73a243f94c1a0a7f5c539936f070d9d41 Mon Sep 17 00:00:00 2001
From: JiaoPL <jplzyc@163.com>
Date: Mon, 13 Nov 2023 15:59:17 +0800
Subject: [PATCH] larger initialize timeout

---
 internlm/utils/timeout.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/internlm/utils/timeout.py b/internlm/utils/timeout.py
index 4e68ce9..c46332f 100644
--- a/internlm/utils/timeout.py
+++ b/internlm/utils/timeout.py
@@ -41,8 +41,8 @@ ENABLE_TIMEOUT = os.getenv("INTERNLM_ENABLE_TIMEOUT", None)
 timeout_threshold_dict = {
     "initialize_distributed_env": 120,
     "nopp_forward_backward_step": 360,
-    "initialize_model": 10,
-    "initialize_optimizer": 20,
+    "initialize_model": 20,
+    "initialize_optimizer": 40,
     "optim_step": 30,
     "get_train_data_loader": 600,
     "get_validation_data_loader": 60,