From f3860f5111939793f29aa90a233fd7f220e8c795 Mon Sep 17 00:00:00 2001 From: chenhaozhe Date: Thu, 15 Oct 2020 17:27:24 +0800 Subject: [PATCH] Fix typo of hccl time out options --- .../ascend_distributed_launcher/get_distribute_pretrain_cmd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 74afd6608a..e53c51c9a5 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()