TensorFlow函數(shù):tf.estimator.FinalExporter

2018-05-03 10:29 更新

tf.estimator.FinalExporter函數(shù)

FinalExporter類

繼承自:Exporter

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

此類最終導(dǎo)出服務(wù)的圖形和檢查點(diǎn).

此類在訓(xùn)練結(jié)束時(shí)執(zhí)行單個(gè)導(dǎo)出.

屬性

name

方法

__init__

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

創(chuàng)建一個(gè)用于tf.estimator.EvalSpec的Exporter.

函數(shù)參數(shù):

  • name:將在導(dǎo)出路徑中使用的Exporter的唯一名稱.
  • serving_input_receiver_fn:不帶參數(shù)并返回一個(gè)ServingInputReceiver的函數(shù).
  • assets_extra:一個(gè)可選的字典,指定如何填充導(dǎo)出的SavedModel中的assets.extra目錄.每個(gè)密鑰都應(yīng)該提供相對(duì)于assets.extra目錄的目標(biāo)路徑(包括文件名).相應(yīng)的值給出了要復(fù)制的源文件的完整路徑.例如,復(fù)制單個(gè)文件而不重命名它的簡(jiǎn)單情況被指定為{'my_asset_file.txt': '/path/to/my_asset_file.txt'}.
  • as_text:是否以文本格式編寫SavedModel原型,默認(rèn)為False.

可能引發(fā)的異常:

  • ValueError:如果任何參數(shù)無(wú)效,則會(huì)引發(fā)此異常.

export

export(
    estimator,
    export_path,
    checkpoint_path,
    eval_result,
    is_the_final_export
)
以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)