Using existing get_world_size to prevent AttributeError 'torch.distributed' has no attribute 'is_initialized'. (#511)