with g.graph.as_default():
sv = tf.train.Supervisor()
with sv.managed_session(config = tf.ConfigProto(allow_soft_placement=True)) as sess:
sv.saver.restore(sess, tf.train.latest_checkpoint(model_path))
您好,我正在使用Flask建立微服务,并且我解决了一个问题。为了使服务正常运行,必须全局还原使用的模型,但是上面显示了使用tf.train.Supervisor的常用方法,并且多个“ with”块构成了本地还原的模型。
任何人都可以帮助我更改样式并在全球范围内恢复模型。