使用tf.train.Supervisor()来还原没有“ with”块的模型

时间:2018-07-27 05:07:27

标签: python tensorflow

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”块构成了本地还原的模型。

任何人都可以帮助我更改样式并在全球范围内恢复模型。

0 个答案:

没有答案