tf.contrib.eager.inf_callback

View source on GitHub

A specialization of inf_nan_callback that checks for infs only.