tensorflow - 如何在 nan loss 训练中失败？

Question

我试图找到如何在损失变成 nan 后训练失败

我找到了关于如何为 Tensorflow 1 执行此操作的旧帖子 tf.train.NanTensorHook(loss, fail_on_nan_loss=False) 仍然会在 TF1.0 中引发异常

但是我目前正在使用 Tensorflow 2，并且我没有在文档中找到任何关于如何使用第二个版本进行操作的提及...

任何帮助将不胜感激 ...

标签： tensorflowtensorflow2.0

Answer 1

为此，您可以使用TerminateOnNaN回调：

from tensorflow.keras.callbacks import TerminateOnNaN

term = TerminateOnNaN()

model.fit(..., callbacks=[term])

它完全符合您的要求，您可以在此处查看文档。