首页 文章

如何使用两个检查点文件来初始化图形

提问于
浏览
0

我训练了两个模特 . 每个都可以产生有用的代表性功能 . 我想使用这些功能来训练第三个模型 . 这意味着前两个模型可以生成第三个模型的输入 . 我想使用tf.contrib.learn.train()来训练第三个模型 . 问题是如何将两个检查点文件导入前两个模型以初始化权重 . 我知道函数中有init_fn和init_op参数 . 但是,我不知道如何使用多个检查点文件 . 你能帮我吗?谢谢 .

1 回答

  • 0

    你可以使用tf.contrib.framework.init_from_checkpoint .

相关问题