python - TensorFlow 训练批处理作业是否有默认输出目录？

Question

最近，我正在尝试为 TensorFlow 实施培训服务。

我无法在文档中找到有关培训作业的输出目录的任何信息。

TensorFlow 是否使用任何默认目录（我对 TensorFlow 真的一点也不熟悉）？

我还在考虑输出目录可以（可能）被编码到训练脚本中，甚至可以通过 CLI 在应用程序调用中指定的可能性。

有人可以帮忙吗？

标签： pythonpython-3.xtensorflowbatch-processingtensor

Answer 1

如果您使用EstimatorAPI，则默认 output_directory 是通过model_dir参数指定的。如果您不指定一个，则会为您创建一个临时目录。请参阅文档字符串：

model_dir：保存模型参数、图形等的目录。这也可用于将检查点从目录加载到估计器中，以继续训练先前保存的模型。如果是 PathLike 对象，则路径将被解析。如果设置为 None，则将使用 config 中的 model_dir。如果两者都设置，则它们必须相同。如果两者都为 None，则将使用临时目录。