Spec-Zone.ru › TensorFlow 1.15

tf.contrib.framework.assign_from_checkpoint_fn

Возвращает функцию, которая присваивает определенные переменные из контрольной точки.

tf.contrib.framework.assign_from_checkpoint_fn(
    model_path, var_list, ignore_missing_vars=False, reshape_variables=False
)

Если ignore_missing_vars равно True и переменные не найдены в контрольной точке, возвращается None.

Аргументы
model_path Полный путь к контрольной точке модели. Чтобы получить последнюю контрольную точку, используйте model_path = tf.train.latest_checkpoint(checkpoint_dir)
var_list Список объектов Variable или словарь, сопоставляющий имена в контрольной точке соответствующим переменным для инициализации. Если пустой или None, возвращается no_op(), None.
ignore_missing_vars Булево значение. Если True, пропускаются переменные, отсутствующие в контрольной точке, с предупреждением вместо ошибки.
reshape_variables Булево значение. Если True, автоматически изменяет размер переменных, которые имеют другой размер, чем переменные в контрольной точке, но имеют то же количество элементов.
Возвращаемое значение
Функция, которая принимает один аргумент, a tf.compat.v1.Session, который применяет операцию присваивания. Если соответствующие переменные не были найдены в контрольной точке, возвращается None.
Исключения
ValueError Если var_list пустой.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/framework/assign_from_checkpoint_fn

Spec-Zone.ru

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