tf.estimator.LatestExporter
| Просмотреть исходный код на GitHub |
Этот класс регулярно экспортирует служебный граф и контрольные точки.
Наследуется от: Exporter
tf.estimator.LatestExporter(
name,
serving_input_receiver_fn,
assets_extra=None,
as_text=False,
exports_to_keep=5
)
В дополнение к экспорту, этот класс также удаляет устаревшие экспорты.
| Аргументы | |
|---|---|
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. |
exports_to_keep | Количество экспортов для сохранения. Более старые экспорты будут удалены. По умолчанию 5. Установите значение None для отключения удаления устаревших экспортов. |
| Исключения | |
|---|---|
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/LatestExporter