bubbliiiing / centernet-tf2

这是一个centernet-tf2的源码,可以用于训练自己的模型。
MIT License
46 stars 27 forks source link

训练的loss为Nan #1

Closed cmfight closed 3 years ago

cmfight commented 3 years ago

您好!up,在b站上看到了你的centernet教程,学到很多,但是我在尝试自己训练的过程中,显示的loss为nan,使用的数据集是voc,输出的信息如下:

1127/1127 [==============================] - 252s 224ms/step - loss: nan - val_loss: nan - lr: 2.5000e-05 Epoch 56/100 1127/1127 [==============================] - ETA: 0s - loss: nan 1127/1127 [==============================] - 252s 224ms/step - loss: nan - val_loss: nan - lr: 2.5000e-05 Epoch 00056: early stopping

cmfight commented 3 years ago

解决了,原先的卡是3090,可能是精度的原因,换到1080上成功执行,感谢up!