如何解决[GluonCV][mxnet]更快的RCNN模型MXNetError:cudaMalloc重试失败:内存不足
我正在 colab 上使用 faster_rcnn_resnet50_v1b_voc 训练我的数据集。我关注了this tutorial,这里是my code。
总的来说,我在 Finetune a pretrained detection model 之后更改了网络初始化:
net = gcv.model_zoo.get_model('faster_rcnn_resnet50_v1b_coco',预训练=真)
net.reset_class(['c'])
并更改了训练/验证数据集
这是我在 Colab 上安装环境的方法
pip install mxnet==1.7.0.post1
pip install mxnet-cu101
pip install gluoncv==0.8.0
错误信息:
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。