如何加载两种不同的张量流模型



我想知道加载两种不同的张量流模型。例如,一个用于检测,另一个用于分类,两个模型需要加载到一个程序中。这两个模型都使用

saver = tf.train.Saver()
saver.restore(isess, ckpt_filename)

加载。但是错误发生了。有没有人成功完成这项工作?

您可以为不同的模型创建不同的图形。因为当您从检查点还原模型时,模型将存储在默认图形中,这就是为什么当您还原不同的模型时会导致错误。

最新更新