(Tensorflow服务)如何使用自定义目录的激励器export_savedmodel?

时间:2018-11-05 04:37:05

标签: tensorflow tensorflow-serving tensorflow-estimator

我想问一下如何为此功能设置目录名称。

    export_dir_base = './export/'
    estimator.export_savedmodel(
        export_dir_base=export_dir_base,
        serving_input_receiver_fn=serving_input_receiver_fn)

当我使用估算器的export_savedmodel函数时,它会为pb文件提供随机整数。

所以,通常看起来像这样,

export / 152342451 /变量

export / 152342451 / saved_model.pb

我想知道如何将其更改为自定义目录名称。

谢谢。

0 个答案:

没有答案