TensorFlow从Checkpoint多次还原

时间:2018-11-07 09:07:58

标签: python tensorflow neural-network deep-learning

我有一个Tensorflow检查点,可从该检查点恢复我的模型:

logits, endpoints = mobilenet_v2.mobilenet(images)
saver = tf.train.Saver()
saver.restore(sess, checkpoint)

我需要以某种方式创建此模型的多个实例,这样我可以提供不同的输入并比较输出,例如:

logits1, endpoints1 = mobilenet_v2.mobilenet(images1)
logits2, endpoints2 = mobilenet_v2.mobilenet(images2)
saver.restore(sess, checkpoint)

,然后我想稍后在代码中使用endpoints1endpoints2。当然,这失败了,因为变量尝试被初始化两次(我想)。

有没有办法实现这一目标?

0 个答案:

没有答案