tf.keras.callbacks.TerminateOnNaN

Class TerminateOnNaN

Inherits From: Callback

Defined in tensorflow/python/keras/callbacks.py.

Callback that terminates training when a NaN loss is encountered.

__init__

__init__()

Initialize self. See help(type(self)) for accurate signature.

Methods

tf.keras.callbacks.TerminateOnNaN.on_batch_begin

on_batch_begin(
    batch,
    logs=None
)

tf.keras.callbacks.TerminateOnNaN.on_batch_end

on_batch_end(
    batch,
    logs=None
)

tf.keras.callbacks.TerminateOnNaN.on_epoch_begin

on_epoch_begin(
    epoch,
    logs=None
)

tf.keras.callbacks.TerminateOnNaN.on_epoch_end

on_epoch_end(
    epoch,
    logs=None
)

tf.keras.callbacks.TerminateOnNaN.on_train_batch_begin

on_train_batch_begin(
    batch,
    logs=None
)

tf.keras.callbacks.TerminateOnNaN.on_train_batch_end

on_train_batch_end(
    batch,
    logs=None
)

tf.keras.callbacks.TerminateOnNaN.on_train_begin

on_train_begin(logs=None)

tf.keras.callbacks.TerminateOnNaN.on_train_end

on_train_end(logs=None)

tf.keras.callbacks.TerminateOnNaN.set_model

set_model(model)

tf.keras.callbacks.TerminateOnNaN.set_params

set_params(params)