Fix typo of hccl time out options

This commit is contained in:
chenhaozhe 2020-10-15 17:27:24 +08:00
parent d020ed1eec
commit f3860f5111
1 changed files with 1 additions and 1 deletions

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