make_extension(trigger=None, default_name=None, priority=None, finalizer=None, initializer=None, on_error=None, **kwargs)¶
Decorator to make given functions into trainer extensions.
This decorator just adds some attributes to a given function. The value of the attributes are given by the arguments of this decorator.
Extensionfor details of trainer extensions. Most of the default values of arguments also follow those for this class.
trigger – Default trigger of the extension.
default_name – Default name of the extension. The name of a given function is used by default.
priority (int) – Default priority of the extension.
finalizer – Finalizer function of this extension. It is called at the end of the training loop.
initializer – Initializer function of this extension. It is called at the beginning of the training loop.
on_error – Error handler callback function of this extension. It is called after an error is raised during the trainer loop.