使用python mnist_tpu.py --use_tpu = false --tpu =在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 它将起作用。