pytorch_pfn_extras.distributed#
Functions
|
Returns a indices of a dataset to be used for the current process. |
|
Initialize torch.distributed environments with values taken from OpenMPI. |
Classes
|
Distributed sampler without duplication |
Site Navigation
Section Navigation
Functions
|
Returns a indices of a dataset to be used for the current process. |
|
Initialize torch.distributed environments with values taken from OpenMPI. |
Classes
|
Distributed sampler without duplication |