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