yolov3-channel-and-layer-pruning icon indicating copy to clipboard operation
yolov3-channel-and-layer-pruning copied to clipboard

Published 20 hours ago • verified publisher icon tanluren

yolov4-tiny.cfg RuntimeError: shape '[256, 384, 3, 3]' is invalid for input of size 875238

Open RnYngxx opened this issue 4 years ago • 1 comments

The following error occurred when I use yolov4-tiny.cfg: Traceback (most recent call last): File "/Users/ren_yx/PycharmProjects/Object detection/YOLO_pruning/train.py", line 542, in train() # train normally File "/Users/ren_yx/PycharmProjects/Object detection/YOLO_pruning/train.py", line 143, in train cutoff = load_darknet_weights(model, weights) File "/Users/ren_yx/PycharmProjects/Object detection/YOLO_pruning/models.py", line 353, in load_darknet_weights conv_w = torch.from_numpy(weights[ptr:ptr + num_w]).view_as(conv_layer.weight) RuntimeError: shape '[256, 384, 3, 3]' is invalid for input of size 875238

RnYngxx avatar Jul 13 '20 05:07 RnYngxx

上传的cfg不对，已更新

tanluren avatar Jul 14 '20 15:07 tanluren