[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

错误信息:

enter image description here

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...