Fix setup_dist (#806)

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

View File

@ -22,7 +22,7 @@ from torch import distributed as dist
def setup_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. rank and world_size are used only if use_ddp_launch is False.