为SavedModel设置张量值

时间:2018-11-21 20:54:04

标签: python tensorflow

我想使用tensorflow SavedModel api保存一个模型,以后可以进行一般部署。但是我的模型使用了dropout,当我实际使用模型进行预测时,我想禁用它。该变量名为keep_probability(预测时我想将其设置为1)。 tensorflow saved_model.simple_save似乎不可能,而且我不知道如何为他们提供https://www.tensorflow.org/api_docs/python/tf/saved_model/builder

我想做什么(澄清我的问题):

simple_save(session,
            export_dir,
            inputs={"x": x},
            static={keep_probability: 1},
            outputs={"z": z})

0 个答案:

没有答案