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

来源：6-22 TensorFlow-ssd 模型训练-实操（2）

weixin_宝慕林202127

2019-08-02

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

写回答

1回答

会写代码的好厨师

2019-08-15

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

weixin_宝慕林202127

谢谢老师回答

2019-08-25

共1条回复

Python3+TensorFlow打造人脸识别智能小程序

理论与实战项目双管齐下，让AI技术真正落地应用，适合毕设展示。

1085 学习 · 538 问题

相似问题

回答 1

回答 3

回答 1

回答 1

回答 1