Spec-Zone.ru › TensorFlow 1.15

tf.train.warm_start

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

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.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 [Необязательно] Один из следующих вариантов:
  • Регулярное выражение (строка), которое определяет, какие переменные нужно инициализировать (см. tf.compat.v1.get_collection). Это выражение будет учитывать только переменные в коллекции TRAINABLE_VARIABLES — если вам нужно инициализировать не-TRAINABLE переменные (например, накопители оптимизатора или статистику нормализации по партиям), используйте нижеприведенный вариант.
  • Список строк, каждая из которых — это регулярное выражение, предоставляемое tf.compat.v1.get_collection для GLOBAL_VARIABLES (см. tf.compat.v1.get_collection). По соображениям обратной совместимости, этот вариант отделён от варианта с единственной строкой.
  • Список переменных для инициализации. Если у вас нет доступа к объектам Variable в месте вызова, используйте вышеуказанный вариант.
  • None, в этом случае будут инициализированы только TRAINABLE-переменные, указанные в 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 для имён переменных, которые не используются. Это обеспечивает более строгую проверку конфигурации переменных, чем полагание на проверку пользователем журналов.

© 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

Spec-Zone.ru

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