Spec-Zone.ru › TensorFlow

tf.saved_model.LoadOptions

Параметры для загрузки SavedModel.

tf.saved_model.LoadOptions(
    allow_partial_checkpoint=False,
    experimental_io_device=None,
    experimental_skip_checkpoint=False,
    experimental_variable_policy=None,
    experimental_load_function_aliases=False
)

Используется в блокнотах

Используется в учебниках
  • Сохранение и загрузка модели с использованием стратегии распределения

Эта функция может быть использована в аргументе options в функциях, которые загружают SavedModel (tf.saved_model.load, tf.keras.models.load_model).

Аргументы
allow_partial_checkpoint bool. По умолчанию False. При включении позволяет SavedModel контрольной точке не полностью соответствовать загруженному объекту.
experimental_io_device строка. Применяется в распределённой среде. Устройство Tensorflow, используемое для доступа к файловой системе. Если None (по умолчанию), то для каждой переменной доступ к файловой системе осуществляется с устройства CPU:0 хоста, на котором эта переменная назначена. Если указано, то доступ к файловой системе для всех переменных осуществляется с этого устройства. Это полезно, например, если вы хотите загрузить из локального каталога, такого как "/tmp", при выполнении в распределённой среде. В этом случае укажите устройство для хоста, на котором доступен каталог "/tmp".
experimental_skip_checkpoint bool. По умолчанию False. Если установлено в True, контрольные точки не будут восстановлены. Обратите внимание, что в большинстве случаев это приведёт к созданию непригодной модели.
experimental_variable_policy строка. Политика, применяемая к переменным при загрузке. Это либо экземпляр перечисления saved_model.experimental.VariablePolicy, либо одна из его строковых значений (регистр не важен). Подробности см. в документации по этому перечислению. Значение None соответствует политике по умолчанию.
experimental_load_function_aliases bool. По умолчанию False. Если установлено в True, атрибут function_aliases будет добавлен к загруженному объекту SavedModel.
Атрибуты
allow_partial_checkpoint
experimental_io_device
experimental_load_function_aliases
experimental_skip_checkpoint
experimental_variable_policy

© 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/api_docs/python/tf/saved_model/LoadOptions

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API