tf.saved_model.SaveOptions
Параметры сохранения в SavedModel.
tf.saved_model.SaveOptions(
namespace_whitelist=None, save_debug_info=False, function_aliases=None,
experimental_io_device=None
)
Эта функция может использоваться в аргументе options в функциях, которые сохраняют SavedModel (tf.saved_model.save, tf.keras.models.save_model).
| Аргументы | |
|---|---|
namespace_whitelist | Список строк, содержащих пространства имен операторов, которые нужно добавить в белый список при сохранении модели. При сохранении объекта, использующего именованные операторы, необходимо явно добавить все пространства имен в белый список. Именованные операторы должны быть зарегистрированы в рамках при загрузке SavedModel. |
save_debug_info | Логическое значение, указывающее, сохраняется ли отладочная информация. Если True, то будет записан файл debug/saved_model_debug_info.pb с содержимым двоичного протокольного буфера GraphDebugInfo, содержащего информацию о трассировке стека для всех операторов и функций, которые сохраняются. |
function_aliases | Словарь Python. Сопоставление строки с объектом, возвращаемым @tf.function. Одна tf.function может генерировать множество ConcreteFunctions. Если инструмент нижнего уровня хочет сослаться на все конкретные функции, сгенерированные одной tf.function, можно использовать аргумент function_aliases для хранения сопоставления имени псевдонима со всеми именами конкретных функций. Например, class MyModel:
@tf.function
def func():
...
@tf.function
def serve():
...
func()
model = MyModel()
signatures = {
'serving_default': model.serve.get_concrete_function(),
}
options = tf.saved_model.SaveOptions(function_aliases={
'my_func': func,
})
tf.saved_model.save(model, export_dir, signatures, options)
|
experimental_io_device | Строка. Применяется в распределенной среде. Устройство TensorFlow, используемое для доступа к файловой системе. Если None (по умолчанию), то для каждой переменной доступ к файловой системе осуществляется с устройства CPU:0 хоста, на котором назначена эта переменная. Если указано, доступ к файловой системе для всех переменных осуществляется с этого устройства. Это полезно, например, если вы хотите сохранить в локальный каталог, такой как "/tmp", при работе в распределенной среде. В этом случае укажите устройство хоста, с которого доступен каталог "/tmp". |
Переменные класса
-
experimental_io_device -
function_aliases -
namespace_whitelist -
save_debug_info
© 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/saved_model/SaveOptions
