Spec-Zone.ru › TensorFlow

tf.compat.v1.train.warm_start

Инициализирует модель с использованием заданных параметров.

tf.compat.v1.train.warm_start(
    ckpt_to_initialize_from,
    vars_to_warm_start='.*',
    var_name_to_vocab_info=None,
    var_name_to_prev_var_name=None
)

Если вы используете tf.estimator.Estimator, эта функция будет автоматически вызвана во время обучения.

Аргументы
ckpt_to_initialize_from [Обязательный] Строка, указывающая каталог с файлами контрольных точек или путь к контрольной точке, из которой нужно инициализировать параметры модели.
vars_to_warm_start [Необязательный] Один из следующих вариантов:
  • Регулярное выражение (строка), задающее переменные для инициализации (см. tf.compat.v1.get_collection). Это выражение будет учитывать только переменные из коллекции TRAINABLE_VARIABLES. Если необходимо инициализировать нетренируемые переменные (например, аккумуляторы оптимизатора или статистики пакетной нормализации), используйте следующий вариант.

  • Список строк, каждая из которых представляет собой область видимости, соответствующую регулярному выражению, передаваемую в tf.compat.v1.get_collection с GLOBAL_VARIABLES (см. tf.compat.v1.get_collection). Для обеспечения обратной совместимости этот вариант отделён от аргумента типа «одна строка».

  • Список переменных для инициализации. Если в месте вызова у вас нет доступа к объектам Variable, используйте предыдущий вариант.

  • None; в этом случае будут инициализированы только тренируемые переменные, указанные в var_name_to_vocab_info.

По умолчанию используется '.*', при котором инициализируются все переменные из коллекции TRAINABLE_VARIABLES. Обратите внимание, что при этом исключаются такие переменные, как аккумуляторы и скользящие статистики пакетной нормализации.

var_name_to_vocab_info [Необязательный] Словарь, сопоставляющий имена переменных (строки) с tf.estimator.VocabInfo. Имена переменных должны соответствовать «полным» переменным, а не именам разделов. Если значение не задано явно, предполагается, что словарь переменной не изменялся.
var_name_to_prev_var_name [Необязательный] Словарь, сопоставляющий имена переменных (строки) с именами ранее обученных переменных в ckpt_to_initialize_from. Если значение не задано явно, предполагается, что имя переменной совпадает в предыдущей контрольной точке и текущей модели. Обратите внимание, что это не влияет на набор инициализируемых переменных и управляет только сопоставлением имён (используйте vars_to_warm_start, чтобы задать переменные для инициализации).
Исключения
ValueError Если WarmStartSettings содержит настройку prev_var_name или VocabInfo для имён переменных, которые не используются. Это позволяет надёжнее проверять конфигурацию переменных, не полагаясь на то, что пользователи будут изучать журналы.

© 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/compat/v1/train/warm_start

Spec-Zone.ru

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