pytorch_pfn_extras.runtime

Classes

pytorch_pfn_extras.runtime.BaseRuntime(...)	A base class for collections of device-specific callback functions.
pytorch_pfn_extras.runtime.PyTorchRuntime(...)	A collections of callback functions for the devices that PyTorch supports by default.