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 , если экспорт пропущен. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/estimator/FinalExporter