keras_benchmarks of tensorflow for data_generator的相关库是什么？

Question

我尝试运行keras_benchmarks of tensorflow，因为我遇到了与https://github.com/tensorflow/benchmarks/issues/91类似的问题。

当我在基准/脚本/keras_run_benchmark.py中运行"python run_benchmark.py“时，

python告诉我：

ImportError:没有名为data_generator的模块

我已经安装并运行了与tf_cnn_benchmarks相关的东西，并且可以成功地对tf_cnn_benchmarks的所有模型进行真正的数据培训，但我仍然不知道data_generator在哪里。

这是我的环境：

Ubuntu 16.04

CUDA 9.0/CuDNN 7.0

python 2.7

PS。我在网上搜索了一下，发现tensorflow有一个名为"tensor2tensor“的库，但是当我安装和导入它时，它并不起作用: python仍然无法在tensor2tensor的data_generators中找到generate_img_input_data。(它被命名为"data_generators"，而不是"data_generator")

从data_generators导入generate_img_input_data ImportError:无法导入名称generate_img_input_data

关于方向我说得对吗？或者还有其他我应该安装和导入的库？

Answer 1

data_generator.py文件在这里，generator.py (它是在您问这个问题的同一天添加的，所以我想这不是巧合吗？)