我想使用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})