pytorch_pfn_extras.training.IgniteExtensionsManager¶
- class pytorch_pfn_extras.training.IgniteExtensionsManager(engine, models, optimizers, max_epochs, *, extensions=None, out_dir='result', writer=None, enable_profile=False)¶
Manages extensions and the current status in Ignite training loop.
- Parameters
engine (ignite.engine.Engine) – Ignite trainer engine
models (dict or torch.nn.Module) – Map of string to Module or an actual Module
optimizers (dict or torch.Optimizer) – Map of string to Optimizer or an actual Optimizer.
max_epochs (int) – Number of epochs in the whole training loop.
extensions (list or None) – List of Extentions to be used.
out_dir (str) – Output directory (default:
result
).writer (writing.Writer object) – Writer that can be used by extensions to write data to custom filesystems.
enable_profile (bool) – Flag to enable/disable profiling of iterations. Default is False.
- Return type
None
- __init__(engine, models, optimizers, max_epochs, *, extensions=None, out_dir='result', writer=None, enable_profile=False)¶
- Parameters
engine (ignite.engine.Engine) –
models (Union[torch.nn.modules.module.Module, Mapping[str, torch.nn.modules.module.Module]]) –
optimizers (Union[torch.optim.optimizer.Optimizer, Mapping[str, torch.optim.optimizer.Optimizer]]) –
max_epochs (int) –
extensions (Optional[Sequence[extension_module.ExtensionLike]]) –
out_dir (str) –
writer (Optional[pytorch_pfn_extras.writing._writer_base.Writer]) –
enable_profile (bool) –
- Return type
None
Methods
__init__
(engine, models, optimizers, …[, …])extend
(extension[, name, trigger, priority, …])Registers an extension to the manager.
get_extension
(name)Returns the extension of a given name.
load_state_dict
(to_load)needs_model_state
([iteration])needs_state_this_iteration
()run_extensions
(*[, completed, only_iterations])set_ignite_handlers
()start_extensions
()state_dict
()Attributes
elapsed_time
epoch
epoch_detail
is_before_training
iteration
models
optimizers
out
raw_models
stop_trigger
updater