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