gigl.common.utils.get_rank

gigl.common.utils.torch_training.get_rank() → int

This is automatically set by Kubeflow PyTorchJob launcher Returns:

int: The index of the process involved in distributed training