Spec-Zone.ru › TensorFlow 1.15

tf.contrib.framework.assign_from_checkpoint

Создаёт операцию для присвоения определённых переменных из контрольной точки.

tf.contrib.framework.assign_from_checkpoint(
    model_path, var_list, ignore_missing_vars=False
)
Аргументы
model_path Полный путь к контрольной точке модели. Чтобы получить последнюю контрольную точку, используйте model_path = tf.train.latest_checkpoint(checkpoint_dir)
var_list Список (возможно, разделённых) Variable объектов или словарь, сопоставляющий имена в контрольной точке соответствующим переменным или списку переменных, которые необходимо инициализировать из значения контрольной точки. Для разделённых переменных имя в контрольной точке должно быть полным именем переменной, а не именем разделённой переменной, например, "my_var", а не "my_var/part_4". Если пусто, возвращает no_op(), {}.
ignore_missing_vars Булево значение. Если True, игнорирует переменные, отсутствующие в контрольной точке, выводит предупреждение вместо завершения работы с ошибкой.
Возвращаемые значения
операцию восстановления restore_op и словарь feed_dict, которые необходимо выполнить для восстановления var_list.
Исключения
ValueError Если ignore_missing_vars имеет значение False и контрольная точка, указанная в model_path, отсутствует для одной из переменных в 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

Spec-Zone.ru

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