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

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(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 current model is torch.Size([2, 1024]).
        size mismatch for roi_heads.box.predictor.cls_score.bias: copying a param with shape torch.Size([6]) from checkpoint, the shape in current model is torch.Size([2]).
        size mismatch for roi_heads.box.predictor.bbox_pred.weight: copying a param with shape torch.Size([24, 1024]) from checkpoint, the shape in current model is torch.Size([8, 1024]).
        size mismatch for roi_heads.box.predictor.bbox_pred.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([8]).
        size mismatch for roi_heads.mask.predictor.mask_fcn_logits.weight: copying a param with shape torch.Size([6, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([2, 256, 1, 1]). 
        size mismatch for roi_heads.mask.predictor.mask_fcn_logits.bias: copying a param with shape torch.Size([6]) from checkpoint, the shape in current model is torch.Size([2]).

为什么我换了数据集,改了defaults.py里面的NUM_CLASS,但是还是有问题呢,仔细一想一开始一怒之下我删除了之前运行时生成的pth
在这里插入图片描述
然后报了新的错误:

FileNotFoundError: [Errno 2] No such file or directory: '.\\model_0001400.pth'

双shfit查了查到底哪里在用这个发现了这个文件
在这里插入图片描述
删除时警告我说有些地方在用,目测是在运行新的数据集时使用了旧的网络参数
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值