KeyError: ‘conv1.1.0.weight‘

最新推荐文章于 2024-05-24 15:45:15 发布

大果粒多

最新推荐文章于 2024-05-24 15:45:15 发布

阅读量3.3k

点赞数

文章标签：深度学习

本文链接：https://blog.csdn.net/weixin_44905678/article/details/120537706

版权

最近在训练faster rcnn +resnet101时，出现了很多问题。
先是训练VOC2007数据集，训练了一周没有框。后来发现原因是载入权值出现了问题，应该这样去载入。

def resnet101(pretrained=False, **kwargs):
    model = ResNet(Bottleneck, [3, 4, 23, 3], **kwargs)
    if pretrained:
        model.load_state_dict(model_zoo.load_url(model_urls['resnet101'<

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

大果粒多

关注关注

0
点赞
踩
4

收藏

觉得还不错? 一键收藏
0
评论
KeyError: ‘conv1.1.0.weight‘

KeyError: ‘conv1.1.0.weight‘最近在训练faster rcnn +resnet101时，出现了很多问题。先是训练VOC2007数据集，训练了一周没有框。后来发现原因是载入权值出现了问题，应该这样去载入。def resnet101(pretrained=False, **kwargs): model = ResNet(Bottleneck, [3, 4, 23, 3], **kwargs) if pretrained: model.load_st
复制链接

扫一扫