pytorch_pfn_extras.training.extensions.accumulate.standard_deviation_accumulate.StandardDeviationAccumulate#
- class pytorch_pfn_extras.training.extensions.accumulate.standard_deviation_accumulate.StandardDeviationAccumulate(conversion_key_pair, trigger=(1, 'epoch'), distributed=False)#
Bases:
AccumulateBase
Methods
__init__
(conversion_key_pair[, trigger, ...])finalize
(manager)Finalizes the extension.
initialize
(manager)Initializes up the manager state.
load_state_dict
(to_load)on_error
(manager, exc, tb)Handles the error raised during training before finalization.
state_dict
()Serializes the extension state.
Attributes
default_name
Default name of the extension.
is_async
name
needs_model_state
priority
trigger
- Parameters:
conversion_key_pair (Tuple[str, str]) –
trigger (Optional[Union[Trigger, Callable[[ExtensionsManagerProtocol], bool], Tuple[float, str]]]) –
distributed (bool) –