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