Browse Source

Fix typo of hccl time out options

tags/v1.1.0
chenhaozhe 5 years ago
parent
commit
f3860f5111
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/nlp/bert/scripts/ascend_distributed_launcher/get_distribute_pretrain_cmd.py

+ 1
- 1
model_zoo/official/nlp/bert/scripts/ascend_distributed_launcher/get_distribute_pretrain_cmd.py View File

@@ -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()


Loading…
Cancel
Save