刚开始直接采用调用一个模型的方法:
(1)定义网络
(2)新建sess:sess = tf.Session(config=config)
(3)定义saver:saver = tf.train.Saver()
(4)导入权重:saver.restore(sess, xxx)
但是,如果在一个项目中同时导入多个模型,会报错,应该是graph冲突,所以需要给每个模型单独新建graph:
g1 = tf.Graph()
isess = tf.Session(graph=g1)
with g1.as_default():
(定义网络模型结构)
isess.run(tf.global_variables_initializer())
saver = tf.train.Saver()
saver.restore(isess, xxx)