torchfoo.distributed.get_world_size#

torchfoo.distributed.get_world_size()[source]#

Get number of distributed processes

Returns:

Number of distributed processes, or 1 if not in distributed mode.

Return type:

int