tf.estimator.FinalExporter
| Просмотреть исходный код на GitHub |
Этот класс экспортирует служебную граф и контрольные точки в конце.
Наследуется от: Exporter
tf.estimator.FinalExporter(
name, serving_input_receiver_fn, assets_extra=None, as_text=False
)
Этот класс выполняет одно экспортное действие в конце обучения.
| Аргументы | |
|---|---|
name | уникальное имя этого Exporter, которое будет использоваться в пути экспорта. |
serving_input_receiver_fn | функция без аргументов, возвращающая ServingInputReceiver. |
assets_extra | Необязательный словарь, определяющий, как заполнить каталог assets.extra в экспортированном SavedModel. Каждый ключ должен указывать путь назначения (включая имя файла) относительно каталога assets.extra. Соответствующее значение должно указывать полный путь исходного файла, который будет скопирован. Например, простой случай копирования одного файла без переименования задаётся как {'my_asset_file.txt': '/path/to/my_asset_file.txt'}. |
as_text | нужно ли записать протокол SavedModel в текстовом формате. По умолчанию False. |
| Исключения | |
|---|---|
ValueError | если какой-либо аргумент некорректен. |
| Атрибуты | |
|---|---|
name | Имя каталога. Имя каталога в каталоге экспорта, в котором записываются экспорты данного типа. Не должно быть |
Методы
export
export(
estimator, export_path, checkpoint_path, eval_result, is_the_final_export
)
Экспортирует указанный Estimator в определённый формат.
| Аргументы | |
|---|---|
estimator | экспортируемый Estimator. |
export_path | строка, содержащая каталог, в который нужно записать экспорт. |
checkpoint_path | путь к контрольной точке для экспорта. |
eval_result | результат выполнения Estimator.evaluate на этой контрольной точке. |
is_the_final_export | Этот булево значение равно True, когда это экспорт в конце обучения. Оно равно False для промежуточных экспортов во время обучения. При передаче Exporter в tf.estimator.train_and_evaluate is_the_final_export всегда равно False, если TrainSpec.max_steps равно None. |
| Возвращаемое значение | |
|---|---|
Строка с путём к экспортированной директории или None если экспорт пропущен. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/estimator/FinalExporter