pytorch_pfn_extras.training.ExtensionsManagerProtocol#

class pytorch_pfn_extras.training.ExtensionsManagerProtocol(*args, **kwargs)#

Bases: Protocol

Methods

`__init__`(args, *kwargs)
`get_extension`(name)

Attributes

`elapsed_time`
`epoch`
`epoch_detail`
`is_before_training`
`iteration`
`models`
`observation`
`optimizers`
`out`
`raw_models`
`reporter`
`stop_trigger`
`writer`

__init__(*args, **kwargs)#

property elapsed_time: float#

property epoch: int#

property epoch_detail: float#

get_extension(name)#

Parameters:: name (str) –
Return type:: Extension

property is_before_training: bool#

property iteration: int#

property models: Mapping[str, Module]#

property observation: reporting.Observation#

property optimizers: Mapping[str, Optimizer]#

property out: str#

property raw_models: Mapping[str, Module]#

property reporter: reporting.Reporter#

property stop_trigger: bool#

property writer: Optional[writing.Writer]#