pytorch_pfn_extras.nn.parallel.distributed#
Functions
|
@contextmanager decorator. |
|
|
Classes
|
Module for distributed data parallelism |
Dictionary that remembers insertion order |
|
Type variable. |
|
|
Context manager/function decorator that adds a label to a code block/function when running autograd profiler. |