Hi guys, I got the problem "
Error(s) in loading state_dict for Darknet:
size mismatch for classifier.weight: copying a param with shape torch.Size([7, 512]) from checkpoint, the shape in current model is torch.Size([14455, 512]).
size mismatch for classifier.bias: copying a param with shape torch.Size([7]) from checkpoint, the shape in current model is torch.Size([14455]).
" while I run the demo.py,
Where is the location of 14455?
Hi guys, I got the problem " Error(s) in loading state_dict for Darknet: size mismatch for classifier.weight: copying a param with shape torch.Size([7, 512]) from checkpoint, the shape in current model is torch.Size([14455, 512]). size mismatch for classifier.bias: copying a param with shape torch.Size([7]) from checkpoint, the shape in current model is torch.Size([14455]). " while I run the demo.py, Where is the location of 14455?