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, если экспорт пропущен. |
© 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/LatestExporter