tf.contrib.tpu.export_estimator_savedmodel
Экспорт обученной модели для вывода на TPU.
tf.contrib.tpu.export_estimator_savedmodel(
estimator, export_dir_base, serving_input_receiver_fn, assets_extra=None,
as_text=False, checkpoint_path=None
)
| Аргументы | |
|---|---|
estimator | Estimator, с помощью которой была обучена модель. |
export_dir_base | Строка, содержащая директорию, в которой будут созданы поддиректории с отметкой времени, содержащие экспортированные SavedModel. |
serving_input_receiver_fn | Функция, которая не принимает аргументов и возвращает ServingInputReceiver или TensorServingInputReceiver. |
assets_extra | Словарь, определяющий, как заполнить директорию assets.extra экспортированного SavedModel, или None, если дополнительные ресурсы не требуются. |
as_text | Флаг, указывающий, следует ли записать протокол SavedModel в текстовом формате. |
checkpoint_path | Путь к контрольной точке для экспорта. Если 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/r1.15/api_docs/python/tf/contrib/tpu/export_estimator_savedmodel