!7351 fix typos in hcc-time-out options

Merge pull request !7351 from chenhaozhe/fix-hccl_time_out-typos
4 years ago · a7e0ffbcf9
parent d5fad7804e f3860f5111
commit a7e0ffbcf9
1 changed files with 1 additions and 1 deletions
--- 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()