模型训练时出现问题

来源:6-22 TensorFlow-ssd 模型训练-实操(2)

微酸袅袅_

2020-02-04

如下图,在终端运行代码之后如下图,笔记本电脑配置挺差的,tensorflow是cpu的,老师有时间的话可以解答一下吗
图片描述

写回答

1回答

会写代码的好厨师

2020-02-10

1. 如果电脑是CPU,检测模型应该没办法训练,除非加大内存,至少保证16G。这样的话训练速度依然很慢。如果是使用CPU的话,建议还是直接跑推理模型。2.其他的模型在改小batchsize的条件下,应该可以训练,只是训练速度仍然会很慢。

0
0

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

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

1085 学习 · 538 问题

查看课程