tf.estimator.FinalExporter
| View source on 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.3/api_docs/python/tf/estimator/FinalExporter