如何在 nan loss 训练中失败？
How to fail during training on nan loss?
我试图找到如何在损失变为 nan 后使训练失败
我找到了有关如何为 Tensorflow 1 执行此操作的旧 post
tf.train.NanTensorHook(loss, fail_on_nan_loss=False) will still raise exception with TF1.0
但我目前正在使用 Tensorflow 2，我没有在文档中找到关于如何使用第二个版本的任何提及...
任何帮助将不胜感激...
为此，您可以使用 TerminateOnNaN 回调：
from tensorflow.keras.callbacks import TerminateOnNaN
term = TerminateOnNaN()
model.fit(..., callbacks=[term])
它完全符合你的要求，你可以查看文档here。
我试图找到如何在损失变为 nan 后使训练失败
我找到了有关如何为 Tensorflow 1 执行此操作的旧 post tf.train.NanTensorHook(loss, fail_on_nan_loss=False) will still raise exception with TF1.0
但我目前正在使用 Tensorflow 2，我没有在文档中找到关于如何使用第二个版本的任何提及...
任何帮助将不胜感激...
为此，您可以使用 TerminateOnNaN 回调：
from tensorflow.keras.callbacks import TerminateOnNaN
term = TerminateOnNaN()
model.fit(..., callbacks=[term])
它完全符合你的要求，你可以查看文档here。