TensorFlow Keras Benchmarks中data_generator相关库及导入报错咨询

首先，你之前尝试tensor2tensor的方向其实走偏了——这个库和TensorFlow Benchmarks项目里的data_generator没有关系，完全是两个独立的东西。咱们回到问题本身来解决：

问题根源 #

你遇到的ImportError本质是Python路径问题：data_generator是TensorFlow Benchmarks项目内部的自定义模块，不是需要额外安装的第三方库。当你在benchmarks/scripts/keras_benchmarks目录下直接执行run_benchmark.py时，Python无法找到上层目录里的模块文件。

而你之前能正常运行tf_cnn_benchmarks，是因为那部分脚本已经处理了路径配置，或者你是在项目根目录下执行的相关命令，Python能识别到内部模块。

具体解决步骤 #

有两种简单的方法可以解决这个路径问题：

方法1：临时设置PYTHONPATH环境变量

在执行脚本前，先把Benchmarks项目的根目录添加到Python的搜索路径中。假设你的Benchmarks项目放在~/tensorflow-benchmarks，执行以下命令：

export PYTHONPATH=/path/to/your/benchmarks:$PYTHONPATH

然后再进入scripts/keras_benchmarks目录运行：

python run_benchmark.py

方法2：修改脚本添加路径

直接在run_benchmark.py的开头添加几行代码，强制Python把项目根目录加入搜索路径：

import sys
import os
# 从当前脚本目录向上回溯到项目根目录
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../../..')))

这样不管你在哪个目录执行脚本，Python都能找到内部的data_generator模块。

额外验证 #

你可以先进入Benchmarks项目的根目录，检查是否存在data_generator.py或者相关的模块目录（通常在scripts/keras_benchmarks或上层的data目录下），确认模块文件确实存在，再用上面的方法解决路径问题即可。

内容的提问来源于stack exchange，提问作者Desolve