tf.estimator.NanTensorHook

Monitors the loss tensor and stops training if loss is NaN.

Inherits From: SessionRunHook

Can either fail with exception or just stop training.

loss_tensor Tensor, the loss tensor.
fail_on_nan_loss bool, whether to raise exception when loss is NaN.

Methods

after_create_session

View source

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.