使用 python mnist_tpu.py --use_tpu=false --tpu='' 在 TPU 中出现错误

Question

我尝试从 tensorflow git 运行这段代码：mnist_tpu with

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 它会起作用的。