tensor dimension mismatch about yolov3-network-slimming HOT 4 OPEN

hi, may be some problem happen in 'darknet53.cfg'? does your cfg have the 'yolo' layer ? i suggest that you can try yolov3.cfg first . And if your cfg have 'yolo' layer, may be try setting 'num=3'

from yolov3-network-slimming.

Thanks, darknet53.cfg is sames as yolov3.cfg(diffrent # of classes).
I changed num to 3. and It works.
But it shows Division by Zero at yolomodel.py line 108.
So i fix this simple problem. Now i have to watch result........

from yolov3-network-slimming.

maybe reduce the '0.5' of lines 105 ‘nProposals = int((pred_conf > 0.5).sum().item())’ in yolomodel.py.

from yolov3-network-slimming.

hello，have you solve this problem? I have the same problem with you.
It always shows "Division by Zero"

from yolov3-network-slimming.