search

ModelTC / United-Perception

United Perception
Apache License 2.0
426 stars 65 forks source link

收集梯度为存在nan #70

Closed TonightGo closed 6 months ago

TonightGo commented 6 months ago

作者你好，感谢你优秀的工作。 我在yolov8实现efl时，用hook来收集梯度, loss值正常，但collect_grad中打印出grad存在nan或者inf值，这个是什么原因？

TonightGo commented 6 months ago

已经解决，交叉熵数值稳定的问题