可以用CPU完成 ssd模型训练吗？

Question

我电脑装的CPU版本的tensorflow 运行model_main.py文件训练的时候，程序就这样挂了。不知道是代码不对，还是CPU、内存不行。电脑配置确实渣。求老师解答。卡了两天了，感谢回答。

Answer 1

会写代码的好厨师 2019-08-15 08:06:09

首先，从这个log上看，不好排查问题。可以先尝试把batchsize改成1，网络用mobilenet，把训练跑来，排查因为内存不够导致的原因。如果搞不定，qq联系我吧。还有就是你的电脑如果是cpu的话，ssd训练要花几天时间了，我不建议自己训练模型。。

• 提问者 weixin_宝慕林202127 #1

  谢谢老师回答