maskrcnn报错：Error(s) in loading state_dict for GeneralizedRCNN

最新推荐文章于 2023-12-12 12:48:24 发布

LifeMaker

最新推荐文章于 2023-12-12 12:48:24 发布

阅读量1k

点赞数 1

文章标签：深度学习 pytorch tensorflow

本文链接：https://blog.csdn.net/qq_41207620/article/details/122224662

版权

RuntimeError 模型加载数据集更改参数形状不匹配 .pth文件

关键词由CSDN通过智能技术生成

RuntimeError: Error(s) in loading state_dict for GeneralizedRCNN:
        size mismatch for roi_heads.box.predictor.cls_score.weight: copying a param with shape torch.Size([6, 1024]) from checkpoint, the shape in cur  File "D:\Anaconda3\envs\maskrcnn_benchmark\lib\site-packages\torch\nn\modules\module.py", line 1045,ite-packages\torch\nn\modules\module.py
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error