利用子类API实现的时候，出现了loss为nan的情况

Question

老师你好，我在利用子类API进行实现的时候，出现了loss为nan的情况。代码和课程是一样的，很奇怪为什么会出现这样的情况？会和tf版本有关么？我tf是2.4.1。谢谢老师。

Answer 1

正十七 2021-05-20 06:45:46

有可能是实现的问题，你把你的实现跟咱们git上的代码对比下看有没有区别？

也有可能是训练的问题，比如梯度爆炸了会导致loss变得不可理喻。这个的话试试调小learning rate.