|View source on GitHub|
Monitors the loss tensor and stops training if loss is NaN.
Compat aliases for migration
See Migration guide for more details.
tf.estimator.NanTensorHook( loss_tensor, fail_on_nan_loss=True )
Can either fail with exception or just stop training.
after_create_session( session, coord )
Called when new TensorFlow session is created.
This is called to signal the hooks that a new session has been created. This
has two essential differences with the situation in which
begin is called:
- When this is called, the graph is finalized and ops can no longer be added to the graph.
- This method will also be called as a result of recovering a wrapped session, not only at the beginning of the overall session.