阅读(9.2k) 书签 (0)

TensorFlow函数:tf.estimator.FinalExporter

2018-05-03 10:29 更新

tf.estimator.FinalExporter函数

FinalExporter类

继承自:Exporter

定义在:tensorflow/python/estimator/exporter.py.

此类最终导出服务的图形和检查点.

此类在训练结束时执行单个导出.

属性

name

方法

__init__

__init__(
    name,
    serving_input_receiver_fn,
    assets_extra=None,
    as_text=False
)

创建一个用于tf.estimator.EvalSpec的Exporter.

函数参数:

  • name:将在导出路径中使用的Exporter的唯一名称.
  • serving_input_receiver_fn:不带参数并返回一个ServingInputReceiver的函数.
  • assets_extra:一个可选的字典,指定如何填充导出的SavedModel中的assets.extra目录.每个密钥都应该提供相对于assets.extra目录的目标路径(包括文件名).相应的值给出了要复制的源文件的完整路径.例如,复制单个文件而不重命名它的简单情况被指定为{'my_asset_file.txt': '/path/to/my_asset_file.txt'}.
  • as_text:是否以文本格式编写SavedModel原型,默认为False.

可能引发的异常:

  • ValueError:如果任何参数无效,则会引发此异常.

export

export(
    estimator,
    export_path,
    checkpoint_path,
    eval_result,
    is_the_final_export
)