mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix setup_dist (#806)

Browse Source
This commit is contained in:
Fangjun Kuang 2023-01-04 12:21:19 +08:00 committed by GitHub
parent 0f26edfde9
commit 8642dbc0bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
icefall/dist.py
View File

@ -22,7 +22,7 @@ from torch import distributed as dist
def setup_dist(
rank, world_size, master_addr=None, master_port=None, use_ddp_launch=False
rank, world_size, master_port=None, use_ddp_launch=False, master_addr=None
):
"""
rank and world_size are used only if use_ddp_launch is False.