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

Question

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

Answer 1

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