YOLOv3-Torch2TRT icon indicating copy to clipboard operation
YOLOv3-Torch2TRT copied to clipboard

Published 20 hours ago • verified publisher icon DocF

is invalid for input of size 1959719

Open menkeyi opened this issue 4 years ago • 0 comments

(openpose) root@ai:~/YOLOv3-Torch2TRT# python3 detect.py Traceback (most recent call last): File "detect.py", line 113, in model_backbone.load_darknet_weights(opt.weights_path) File "/root/YOLOv3-Torch2TRT/models.py", line 365, in load_darknet_weights conv_w = torch.from_numpy(weights[ptr : ptr + num_w]).view_as(conv_layer.weight) RuntimeError: shape '[1024, 512, 3, 3]' is invalid for input of size 1959719

这是啥情况？

menkeyi avatar Feb 20 '21 06:02 menkeyi