如何在 nan loss 训练过程中失败？

Question

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

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

但我目前正在使用 Tensorflow 2，我在文档中没有找到任何关于如何使用第二个版本执行此操作的说明...

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

Answer 1

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

from tensorflow.keras.callbacks import TerminateOnNaN

term = TerminateOnNaN()

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

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