权重与YOLO head 不匹配问题

[SCIKings]

导师，您好！我想问一下，我把yolov4网络做了略微的修改，就出现了以下错误，该如何修改呢？非常期待您的回复，谢谢！RuntimeError: Error(s) in loading state_dict for YoloBody: size mismatch for yolo_head3.1.weight: copying a param with shape torch.Size([75, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([255, 256, 1, 1]). size mismatch for yolo_head3.1.bias: copying a param with shape torch.Size([75]) from checkpoint, the shape in current model is torch.Size([255]).

[sanxincao]

检查一下输入输出，形状不对

[SCIKings]

好的，我去看下，谢谢您的回复！

------------------ 原始邮件 ------------------ 发件人: "careless @.>; 发送时间: 2022年4月12日(星期二) 下午2:45 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [bubbliiiing/yolov4-pytorch] 权重与YOLO head 不匹配问题 (Issue #292)

检查一下输入输出，形状不对

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

[bubbliiiing]

看问题汇总吧