YOLO-pytorch icon indicating copy to clipboard operation
YOLO-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon AceCoooool

weights 转 pth 报错

Open yluo39github opened this issue 5 years ago • 1 comments

Weights Header: [ 0 1 0 32013312] loading the weights ... conv2d bn leaky (32, 3, 3, 3) conv2d bn leaky (64, 32, 3, 3) Traceback (most recent call last): File "tools/yad2t.py", line 124, in weight2pth(config.cfg_path, config.weight_path, config.output_path) File "tools/yad2t.py", line 84, in weight2pth net_dict[keys[key_num + 1]].copy_(torch.from_numpy(bn_weights[0])) RuntimeError: The size of tensor a (3) must match the size of tensor b (64) at non-singleton dimension 3

yluo39github avatar Aug 12 '20 18:08 yluo39github

Same problem

dorianboille avatar Dec 07 '22 09:12 dorianboille