当我尝试使用我自己的图像基于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和图像数据。
非常感谢!
答案 0 :(得分:0)
我自己解决了这个愚蠢的问题。 我买了一台GTX 960,谢谢大家。 我一步一步地重新开始如下: http://blog.twman.org/2016/06/tensorflow.html