chainer.training.extensions.FailOnNonNumber¶
-
class
chainer.training.extensions.
FailOnNonNumber
[source]¶ Trainer extension to raise RuntimeError if parameters contain NaN or Inf.
Although parameters including non-number such as NaN and Inf are unnecessary in most cases,
Trainer
will continue to compute even if the parameters in a given optimizer diverge. This extension is aimed to reduce unnecessary computations by throwingRuntimeError
if the parameters contain NaN or Inf.Methods
-
__call__
(trainer)[source]¶ Invokes the extension.
Implementations should override this operator. This method is called at iterations which the corresponding trigger accepts.
Parameters: trainer (Trainer) – Trainer object that calls this operator.
-
initialize
(trainer)[source]¶ Initializes up the trainer state.
This method is called before entering the training loop. An extension that modifies the state of
Trainer
can override this method to initialize it.When the trainer has been restored from a snapshot, this method has to recover an appropriate part of the state of the trainer.
For example,
ExponentialShift
extension changes the optimizer’s hyperparameter at each invocation. Note that the hyperparameter is not saved to the snapshot; it is the responsibility of the extension to recover the hyperparameter. TheExponentialShift
extension recovers it in itsinitialize
method if it has been loaded from a snapshot, or just setting the initial value otherwise.Parameters: trainer (Trainer) – Trainer object that runs the training loop.
-
serialize
(serializer)[source]¶ Serializes the extension state.
It is called when a trainer that owns this extension is serialized. It serializes nothing by default.
Attributes
-
default_name
¶ Default name of the extension.
It is the name of the class by default. Implementation can override this property, or provide a class attribute to hide it.
-
priority
= 100¶
-
trigger
= (1, 'iteration')¶
-