- 博客(2)
- 收藏
- 关注
原创 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
1680
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人