训练ssd模型时报错

Question

老师您好，我想请问一下，在训练时报下列错误应该怎么办？OOM when allocating tensor with shape[64,128,160,160] and type float。我查了一下网上说是显卡显存容纳不了这么多数据，但是我不知道改哪里的代码。
或者可以提供一下您训练完的frozen_inference_graph.pb文件让我进行一下测试和用于之后的小程序编程嘛
已找到！

Answer 1

会写代码的好厨师 2019-07-25 22:35:15

显卡容量不够的话，首先考虑减小batchsize，然后，在考虑减少主干网络的大小，这个该起来都比较复杂，可以qq联系我，告诉你怎么改。另外，你的显卡是多少？一定要注意的是模型改太小会影响性能。

• 提问者 baoxiang777 #1

  非常感谢！