pytorch_pfn_extras.handler

Functions

pytorch_pfn_extras.handler.forward(block)	Returns a CodeBlock that performs the forward pass for the given torch.nn.Module or another CodeBlock.
pytorch_pfn_extras.handler.torch_autocast([...])
pytorch_pfn_extras.handler.update_parameters(...)	Returns a CodeBlock that performs the forward, backward passes and applies the optimizer step for the given torch.nn.Module or another CodeBlock.

Classes

pytorch_pfn_extras.handler.BaseHandler(...)	Base class of Handler.
pytorch_pfn_extras.handler.BaseLogic([options])
pytorch_pfn_extras.handler.ClousureLogic([...])	A set of methods that defines the training logic.
pytorch_pfn_extras.handler.CodeBlock(func, ...)	Class that is used to specify and apply actions to a callable.
pytorch_pfn_extras.handler.CodeBlockLogic([...])	A set of methods that defines the training logic.
pytorch_pfn_extras.handler.Handler(logic, ...)	A set of callback functions to perform device-specific operations.
pytorch_pfn_extras.handler.Logic([...])	A set of methods that defines the training logic.