tf.saved_model.LoadOptions
Параметры для загрузки SavedModel.
tf.saved_model.LoadOptions(
allow_partial_checkpoint=False,
experimental_io_device=None,
experimental_skip_checkpoint=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, чекпоинты не будут восстановлены. Обратите внимание, что в большинстве случаев это приведёт к созданию непригодной модели. |
| Атрибуты | |
|---|---|
allow_partial_checkpoint | |
experimental_io_device | |
experimental_skip_checkpoint | |
© 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/versions/r2.9/api_docs/python/tf/saved_model/LoadOptions