【问题标题】:How can I implement transfert learning with tensorflow如何使用 tensorflow 实现迁移学习
【发布时间】:2017-05-10 11:32:23
【问题描述】:

我目前正在研究一个深度神经网络,我想对我的模型进行迁移学习,我已经创建了一个从另一个模型生成的 ckpt,我想在我自己的模型中使用这个 ckpt,问题是我的模型完全不同“层数,输出数”,我的问题是如何在最低层使用 ckpt?

我用的是张量流

谢谢

【问题讨论】:

  • 你有其他ckpt的架构吗?
  • 是的,我有架构
  • 你想如何使用它作为你的最低层?想到几个选项,您的模型可以作为输入(仅另一个模型)或(原始数据和另一个模型)。如果您接收原始数据并且如果它的形状与其他模型的输出不同,则您必须执行诸如如何将 wide-and-deep 放在一起的操作。让我知道,我可以提出一些具体的建议。

标签: python-3.x tensorflow deep-learning conv-neural-network


【解决方案1】:

你需要的是tf.train.Saver 类。有var_list 参数,您可以使用它来设置当前图表中保存的权重和变量之间的对应关系。

【讨论】:

    猜你喜欢
    • 2016-11-28
    • 2020-10-31
    • 2020-08-06
    • 2018-11-19
    • 1970-01-01
    • 1970-01-01
    • 2016-05-18
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多