search

bubbliiiing / yolov7-pytorch

这是一个yolov7的库，可以用于训练自己的数据集。
GNU General Public License v3.0
871 stars 151 forks source link

模型使用float16精度预测时，经过前几层卷积就会出现nan，请问是为什么呢 #85

Open TWK2022 opened 1 year ago

TWK2022 commented 1 year ago

19645f6392cfb878fcd8c42fe97784c 9f41e05248fd7a4a3eb28f44453b56d

TWK2022 commented 1 year ago

发现是因为版本问题，已解决，谢谢

bubbliiiing commented 1 year ago

加油