class ModelExportSpec

所在模块 caicloud.clever.tensorflow。

模型导出配置 ModelExportSpec 用于配置要导出的业务模型相关信息。该配置在 model_fn 返回的 ModelFnHandler 对象中指定。

__init__

__init__(
    export_dir, 
    input_tensors=None,
    output_tensors=None,
    features=None,
    assert_collections=None, 
    legacy_init_op=None, 
    main_op=None)

创建一个模型导出配置。

参数：

• export_dir：模型导出的目录路径，不能为空；

• input_tensors：模型输入的别名到实际 tf.Tensor 的字典，例如 {'image': image}；

• output_tensors：模型输出的别名到实际 tf.Tensor 的字典，例如 {'logits': logits}；

• features：用于 tf.contribl.learn 的模型训练任务指定特征向量。

• assert_collections：附加资产文件列表，可选。

  资产文件会在模型导出和导入时被当作模型的一部分进行处理。资产文件主要应用场景：训练模型的某些操作需要外部附加文件进行初始化等。在导出模型的时候，资产文件会被拷贝到模型导出路径的 assets 目录下。

• legacy_init_op：导出模型被恢复模型参数时要被执行的初始化操作，可选。

• main_op：导出模型在被加载时执行的操作，可选。设置了此参数将会忽略 legacy_init_op 参数。