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