我们正在尝试使用TensorFlow示例中的文本分类示例(tensorflow / examples / learn / text_classification.py)。它与db_pedia数据一起很好地工作。

现在,我们尝试使用Saver保存/恢复模型,但是我们没有在哪里使用Saver API作为text_classification中的代码。py根本不使用Session,并且Saver API需要使用session来保存/恢复。

最佳答案

本示例使用tf.estimator.Estimator,它具有特殊的方法
export_savedmodel保存。

另外,您可以在constructor中指定model_dir


  保存模型参数,图形等的目录。也可以是
  用于将目录中的检查点加载到估算器中
  继续训练以前保存的模型。如果为None,则model_dir在
  如果设置,将使用config。如果两者都设置,则必须相同。如果
  两者均为“无”,将使用一个临时目录。

08-24 13:07