Hello guys,
I have been trying to implement the tiny YOLO v3 model and when I try to load the pre-trained weights into the model,I get this error.
File “/home/rahulrampa/.spyder2/Yoloimplementation.py”, line 274, in loadweights
conv_weights=conv_weights.view_as(conv.weight.data)
RuntimeError: shape ‘[512, 256, 3, 3]’ is invalid for input of size 723278
Any ideas on how to fix this problem?
Thanks in advance