loss 变 NaN 使用 hook 逐层定位实操教程|AI 工程师踩坑实录

训练过程 loss 突然变成 NaN,直接调小学习率属于盲猜,最靠谱的办法就是注册 forward hook 逐层监控每一层输出,精准抓到最先出现