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 | [Необязательный] Один из следующих вариантов:
По умолчанию используется |
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