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