YOLO模型train.py文件报错:size mismatch for last_layer0.6.weight: copying a param with shape torch.
YOLO模型train.py文件报错问题在训练YOLO模型后将自己权重用于预测时总是提示如下报错信息:RuntimeError: Error(s) in loading state_dict for YoloBody: size mismatch for last_layer0.6.weight: copying a param with shape torch.Size([24, 1024, 1, 1]) from checkpoint, the shape in current model i
原创
2022-05-19 10:41:43 ·
1757 阅读 ·
2 评论