github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

larger initialize timeout

pull/495/head
JiaoPL 2023-11-13 15:59:17 +08:00
parent 626ed0fc5e
commit 815678f73a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internlm/utils/timeout.py
View File

@ -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,