- 博客(2)
- 收藏
- 关注
RSS订阅原创 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
2021-09-28 21:21:19
3633
原创 RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:50
RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:50 在运行pycharm时突然报这个错误，只知道是环境问题，但是在安装过程中很顺利，但是运行就出错。记录一下自己的解决办法。 1.检查cuda是否安装成功 可以参考下面这个链接。https://zhuanlan.zhihu.com/p/139668028 如果失败了。那就下一步。 2.检查显卡 在设备管理器-显示适配器-
2021-04-13 09:07:17
1221
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人