【问题标题】:Saving wide and deep tensorflow model保存宽而深的张量流模型
【发布时间】:2019-01-26 03:07:10
【问题描述】:

我在玩Wide and Deep learning example from Tensorflow。我想保存训练有素的分类器以供以后用于预测任务,但我真的不知道怎么做。 DNNLinearCombinedClassifier 没有可用的保存方法,并且酸洗对象也会失败。

任何想法如何保存它?

【问题讨论】:

    标签: python tensorflow


    【解决方案1】:

    this doc 的检查点保存部分应该会回答您的问题。

    【讨论】:

      【解决方案2】:

      你在找https://www.tensorflow.org/versions/r0.10/how_tos/variables/index.html#saving-and-restoring吗?

      特别是保存变量恢复变量。此外,您可以使用 Checkpoint Files 定期保存权重(如果您的计算机在中途停止训练时很有用,您不必从头开始)。

      【讨论】:

      • 可能是这样,但不幸的是 Saver 对象需要访问模型未公开的 Session。
      • 您是否尝试过使用model_dir 标志?根据github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/…,model_dir 标志将为您提供:model_dir: Directory to save model parameters, graph and etc. This can also be used to load checkpoints from the directory into a estimator to continue training a previously saved model.,所以我敢打赌,它用于您在深度和广泛的东西中使用的其他分类器。如果是这种情况,那么您可以编写自己的 .py 文件来恢复它并执行您需要做的事情。
      • 是的,设置 model_dir 有效 - 下次我实例化模型时,模型会自动恢复。我期待我需要做更多的事情。谢谢!
      猜你喜欢
      • 2017-06-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2018-07-10
      • 2016-10-26
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多