【发布时间】:2016-07-09 17:44:09
【问题描述】:
当我尝试在 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 训练 inception-v3 模型吗?”
PS:我知道如果只使用 cpu 会比 gpu 花费更多时间,因此我减少了 max_steps 和图像数据。
非常感谢!
【问题讨论】:
标签: tensorflow