Tensorflow同时加载两个网络模型

Tensorflow同时加载两个网络模型

首先贴上我的参考来源 tensorflow中一种融合多个模型的方法

1.使用场景
假设我们有训练好的模型A,B,C，我们希望使用A,B,C中的部分或者全部变量，合成为一个模型D，用于初始化或其他目的，就需要融合多个模型的方法
2.如何实现
我们可以先声明模型D，再创建多个Saver实例，分别从模型A,B,C的保存文件（checkpoint文件）中读取所需的变量值，来达成这一目的，下面是示例代码：
首先创建一个只包含w1,w2两个变量的模型，初始化后保存：
然后再创建一个只包含w2,w3两个变量的模型，也是初始化后保存：
最后我们创建一个包含w1，w2，w3变量的模型，从上面两个保存的ckp文件中恢复
：

def restore_model():
    w1 = tf.get_variable("w1", shape=[3, 1], initializer=tf.truncated_normal_initializer(),trainable=True)
    w2 = tf.get_variable("w2", shape=[3, 1], initializer=tf.truncated_normal_initializer(),trainable=True)
    w3 = tf.get_variable("w3", shape