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

fix monitor config key error when args_check is False

pull/362/head
JiaoPL 2023-09-25 14:16:53 +08:00
parent 64d4159f89
commit 0565e17b8d
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
internlm/initialize/launch.py
View File

@ -468,13 +468,14 @@ def initialize_distributed_env(
args_sanity_check() args_sanity_check()
# init light monitor client # init light monitor client
alert_config = gpc.config.monitor.alert if gpc.config.get("monitor") and gpc.config.monitor.get("alert"):
if alert_config.enable_feishu_alert and gpc.is_rank_for_log(): alert_config = gpc.config.monitor.alert
light_monitor_address = alert_config.light_monitor_address if alert_config.enable_feishu_alert and gpc.is_rank_for_log():
if light_monitor_address: light_monitor_address = alert_config.light_monitor_address
initialize_light_monitor(light_monitor_address) if light_monitor_address:
else: initialize_light_monitor(light_monitor_address)
logger.warning("monitor address is none, monitor could not be used!") else:
logger.warning("monitor address is none, monitor could not be used!")
def get_config_value(config, key, defalut): def get_config_value(config, key, defalut):