pytorch_pfn_extras.training.extensions.MaxAccumulate#

class pytorch_pfn_extras.training.extensions.MaxAccumulate(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) –