tensorflow - Tensorflow - Saved_model_cli.py - （在 VSCode 中使用 Code Runner 编译时出现 ModuleNotFoundError）

Question

我对这个领域有点陌生——希望这个问题的答案对于更有经验的人来说很简单（我有点难过）。

我正在使用 LSTM 构建一个简单的 RNN 机器学习模型，并将该模型保存为 Tensorflow 服务。我使用 keras API 以这种格式保存。现在尝试使用“saved_model_cli.py”检查保存的模型。我正在使用 VSCode 构建模型 - 问题是当我在终端上（在 VSCode 内）运行它时，它处理得很好并且没有错误，但是当使用代码运行器（通过输出窗口）运行它时，我得到一个 modulenotfound 错误? 指定的模块位于“lib”文件夹（在 venv 文件夹中）。它的行为方式是否有原因 - 感谢支持。

代码示例

tf.keras.models.save_model(
    model,
    export_path,
    overwrite=True,
    include_optimizer=True,
    save_format=None,
    signatures=None,
    options=None,
)

print("\nSaved Model Contents:")
with open("saved_model_contents.txt", "w") as f:
    p1 = subprocess.run(["ls", "-al", export_path], stdout=f, text=True)

p2 = subprocess.run(["ls", export_path], stdout=subprocess.PIPE, text=True, check=True)
print(p2.stdout)

with open("saved_model_inspect.txt", "w") as g:
    p3 = subprocess.run(
        "python3 /Users/lisatsakalian/Desktop/Coding_Projects/ML_Projects_Dev/NLP/project_env/lib/python3.9/site-packages/tensorflow/python/tools/saved_model_cli.py  show --dir /var/folders/5y/s74tfz4n3nndnj8gdd2_jdmw0000gn/T/1 --all",
        shell=True,
        stdout=g,
        text=True,
    )

标签： tensorflowkerasvisual-studio-codetensorflow-serving