使用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获得了该文件，它将正常工作。