我在利用TensorFlow Object Detection API预训练模型训练自己的数据时出现:The only thing to do is reduce RAM usage by reducing the batch size or using a smaller model and trying again.
意思是GPU使用率太高了,然后进行新的运算时就崩溃了。
解决方法:
(1)关掉其他应用程序或重启机器,再试一下
(2)减少训练数据量,减少bach_size的值
(3)实在不行,就换一个内存更大GPU