tf.train.warm_start
Инициализирует модель с помощью заданных параметров.
tf.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 для имён переменных, которые не используются. Это обеспечивает более строгую проверку конфигурации переменных, чем полагание на проверку пользователем журналов. |
© 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/train/warm_start