使用 python mnist_tpu.py 在 TPU 中获取错误 --use_tpu=false --tpu=

Question

我尝试从tensorflow git运行此代码： mnist_tpu

python mnist_tpu.py --tpu='' 
python mnist_tpu.py --use_tpu=false --tpu=''

但是我一直收到以下错误：

AttributeError: module 'official.mnist.mnist' has no attribute 'create_model'

我不知道create_model方法在哪里定义。我感觉文档仍然缺少一些信息。

有什么想法吗？提前谢谢。

Answer 1

'create_model' 在 mnist.py 中定义，它与 mnist_tpu.py 一起存在。确认您有来自 github 的文件：https://github.com/tensorflow/models/blob/master/official/mnist/mnist.py 它会起作用。