search

cuixing158 / yolov3-yolov4-matlab

Import and export Darknet™ models within MATLAB deep learning networks.
93 stars 44 forks source link

训练的时候 报错conv_24层不存在 #25

Open 970814 opened 3 years ago

970814 commented 3 years ago

yolov4.weights 和 yolov4-tiny.weights 都不存在 conv_24 层， 下图的第二行代码报错，只存在concat24层而不是 conv_24 层 image

后来我换成 yolov3-tiny.weights ，可以正常训练，不会报这个错，但是在训练过程中报 loss is nan， image

970814 commented 3 years ago

大佬求助

970814 commented 3 years ago

已经 解决，作者实现的代码是针对 yolov3 训练的，yolov4 要自己改，是替换 140，151,162 三个卷积层，但NaN问题依然没有解决

970814 commented 3 years ago

NaN问题依然没有解决,求助，尝试了降低学习率也不行，出现NaN 参数主要是，Prediction Feature 全是NaN