我们正在尝试使用TensorFlow示例中的文本分类示例(tensorflow / examples / learn / text_classification.py)。它适用于db_pedia数据。
现在我们正在尝试使用Saver保存/恢复模型,但是我们没有在哪里使用Saver API,因为text_classification.py中的代码根本不使用Session,Saver API需要会话来保存/恢复。
答案 0 :(得分:0)
此示例使用tf.estimator.Estimator
,它有一个特殊方法
export_savedmodel
用于保存。
此外,您可以在constructor中指定model_dir
:
保存模型参数,图表等的目录。这也可以 用于将目录中的检查点加载到估算器中 继续训练以前保存的模型。如果为None,则为model_dir 如果设置,将使用config。如果两者都设置,它们必须相同。如果 两者都是None,将使用临时目录。