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