diff --git a/model_zoo/official/nlp/bert/scripts/ascend_distributed_launcher/get_distribute_pretrain_cmd.py b/model_zoo/official/nlp/bert/scripts/ascend_distributed_launcher/get_distribute_pretrain_cmd.py index 74afd6608a5..e53c51c9a56 100644 --- a/model_zoo/official/nlp/bert/scripts/ascend_distributed_launcher/get_distribute_pretrain_cmd.py +++ b/model_zoo/official/nlp/bert/scripts/ascend_distributed_launcher/get_distribute_pretrain_cmd.py @@ -77,7 +77,7 @@ def distribute_pretrain(): print("hccl_config_dir:", args.hccl_config_dir) print("hccl_time_out:", args.hccl_time_out) - cmd = append_cmd_env(cmd, 'HCCL_CONNECTION_TIMEOUT', args.hccl_time_out) + cmd = append_cmd_env(cmd, 'HCCL_CONNECT_TIMEOUT', args.hccl_time_out) cmd = append_cmd_env(cmd, 'RANK_TABLE_FILE', args.hccl_config_dir) cores = multiprocessing.cpu_count()