Question

当我尝试使用我自己的图像基于Inception-v3进行训练时如下：

https://github.com/tensorflow/models/blob/master/inception/inception/inception_train.py

我收到此错误：

File "/tensorflow/tensorflow/models/inception/bazel-bin/inception/flowers_train.runfiles/inception/inception_train.py", line 207, in train
    assert FLAGS.batch_size % FLAGS.num_gpus == 0, (
ZeroDivisionError: division by zero

我的问题是“我是否有可能只使用cpu训练初始-v3模型？”

PS：我知道如果只使用比gpu花费更多时间的cpu，我会减少max_steps和图像数据。

非常感谢！

Answer 1

我自己解决了这个愚蠢的问题。我买了一台GTX 960，谢谢大家。我一步一步地重新开始如下： http://blog.twman.org/2016/06/tensorflow.html

我是否有可能只用cpu训练inception-v3模型？

1 个答案: