Spec-Zone.ru › TensorFlow

tf.compat.v1.train.Scaffold

Структура для создания или сбора компонентов, обычно необходимых для обучения модели.

tf.compat.v1.train.Scaffold(
    init_op=None,
    init_feed_dict=None,
    init_fn=None,
    ready_op=None,
    ready_for_local_init_op=None,
    local_init_op=None,
    summary_op=None,
    saver=None,
    copy_from_scaffold=None,
    local_init_feed_dict=None
)

Используется в блокнотах

Используется в руководстве
  • Оценщики
  • Перенос контрольных точек модели

При построении модели для обучения обычно необходимы операции для инициализации переменных, способ для их контрольных точек, операция для сбора данных для визуализатора и т. д.

Различные библиотеки, построенные на основе основной библиотеки TensorFlow, заботятся о создании некоторых или всех этих компонентов и сохранении их в известных коллекциях в графе. Класс Scaffold помогает выбрать эти компоненты из коллекций графа, создавая и добавляя их в коллекции при необходимости.

Если вы вызываете конструктор Scaffold без аргументов, он выберет компоненты из коллекций, создавая значения по умолчанию при необходимости при вызове init_op. Вы можете передать аргументы в конструктор, чтобы предоставить свои собственные компоненты. Компоненты, которые вы передаёте в конструктор, не добавляются в коллекции графа.

Следующие компоненты непосредственно доступны как атрибуты объекта Scaffold:

  • saver: Объект tf.compat.v1.train.Saver, отвечающий за сохранение переменных. По умолчанию выбирается из и сохраняется в коллекцию SAVERS графа.
  • init_op: Операция для выполнения инициализации переменных. По умолчанию выбирается из и сохраняется в коллекцию INIT_OP графа.
  • ready_op: Операция для проверки инициализации переменных. По умолчанию выбирается из и сохраняется в коллекцию READY_OP графа.
  • ready_for_local_init_op: Операция для проверки того, что глобальное состояние было инициализировано, и можно безопасно выполнить local_init_op. По умолчанию выбирается из и сохраняется в коллекцию READY_FOR_LOCAL_INIT_OP графа. Это необходимо, когда инициализация локальных переменных зависит от значений глобальных переменных.
  • local_init_op: Операция для инициализации локальных переменных. По умолчанию выбирается из и сохраняется в коллекцию LOCAL_INIT_OP графа.
  • summary_op: Операция для выполнения и слияния сводок в графе. По умолчанию выбирается из и сохраняется в коллекцию SUMMARY_OP графа.

Вы также можете передать следующие дополнительные компоненты в конструктор:

  • init_feed_dict: Словарь фида сессии, который следует использовать при выполнении операции инициализации.
  • init_fn: Вызываемая функция, которая выполняется после операции инициализации для выполнения дополнительных инициализаций. Функция вызывается как init_fn(scaffold, session).
Аргументы
init_op Необязательная операция для инициализации переменных.
init_feed_dict Необязательный словарь фида сессии для использования при выполнении init_op.
init_fn Необязательная функция для инициализации модели после выполнения init_op. Будет вызвана как init_fn(scaffold, session).
ready_op Необязательная операция для проверки инициализации переменных. Должна возвращать пустой 1D строковый тензор, когда переменные инициализированы, или непустой 1D строковый тензор, перечисляющий имена неинициализированных переменных.
ready_for_local_init_op Необязательная операция для проверки инициализации глобальных переменных и того, что можно выполнить local_init_op. Должна возвращать пустой 1D строковый тензор, когда глобальные переменные инициализированы, или непустой 1D строковый тензор, перечисляющий имена неинициализированных глобальных переменных.
local_init_op Необязательная операция для инициализации локальных переменных.
summary_op Необязательная операция для сбора всех сводок. Должна возвращать скалярный строковый тензор, содержащий сериализованный Summary протокол.
saver Необязательный объект tf.compat.v1.train.Saver для сохранения и восстановления переменных. Также может быть объектом tf.train.Checkpoint, в этом случае сохраняются контрольные точки на основе объектов. Это также загрузит некоторые контрольные точки на основе объектов, сохраненные из других мест, но эта загрузка может быть ненадёжной, так как она использует фиксированные ключи, а не выполняет полное сопоставление на основе графа. Например, если переменная имеет два пути от объекта Checkpoint, потому что два объекта Model используют объект Layer, который ее владеет, удаление одного объекта Model может изменить ключи и нарушить загрузку контрольных точек через этот API, тогда как сопоставление на основе графа сопоставит переменную через другой объект Model.
copy_from_scaffold Необязательный объект Scaffold для копирования полей. Его поля будут перезаписаны предоставленными полями в этой функции.
local_init_feed_dict Необязательный словарь фида сессии для использования при выполнении local_init_op.
Атрибуты
init_feed_dict
init_fn
init_op
local_init_feed_dict
local_init_op
ready_for_local_init_op
ready_op
saver
summary_op

Методы

default_local_init_op

Посмотреть исходный код

@staticmethod
default_local_init_op()

Возвращает операцию, объединяющую операторы инициализации локальных переменных по умолчанию.

Эта операция используется во время инициализации сессии, когда Scaffold инициализируется без указания аргумента local_init_op. Она включает tf.compat.v1.local_variables_initializer, tf.compat.v1.tables_initializer, а также инициализирует локальные ресурсы сессии.

Возвращает
Операция инициализации локальных переменных по умолчанию для Scaffold.

finalize

Посмотреть исходный код

finalize()

Создаёт операции при необходимости и завершает процесс построения графа.

get_or_default

Посмотреть исходный код

@staticmethod
get_or_default(
    arg_name, collection_key, default_constructor
)

Получает из кэша или создаёт операцию по умолчанию.

© 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/Scaffold

Spec-Zone.ru

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