Spec-Zone.ru › TensorFlow

tf.compat.v1.GraphKeys

Стандартные имена для коллекций графа.

Стандартная библиотека использует различные общеизвестные имена для сбора и получения значений, связанных с графом. Например, подклассы tf.Optimizer по умолчанию оптимизируют переменные, собранные в tf.GraphKeys.TRAINABLE_VARIABLES, если не указано иное, но также можно передать явный список переменных.

Определены следующие стандартные ключи:

  • GLOBAL_VARIABLES: коллекция объектов Variable по умолчанию, общая для распределённой среды (переменные модели являются её подмножеством). Подробнее см. в разделе tf.compat.v1.global_variables. Как правило, все переменные TRAINABLE_VARIABLES входят в MODEL_VARIABLES, а все переменные MODEL_VARIABLES — в GLOBAL_VARIABLES.
  • LOCAL_VARIABLES: подмножество объектов Variable, локальных для каждой машины. Обычно используется для временных переменных, например счётчиков.
  • MODEL_VARIABLES: подмножество объектов Variable, используемых в модели для инференса (прямого распространения).
  • TRAINABLE_VARIABLES: подмножество объектов Variable, которые будут обучаться с помощью оптимизатора. Подробнее см. в разделе tf.compat.v1.trainable_variables.
  • SUMMARIES: объекты Tensor, созданные в графе. Подробнее см. в разделе tf.compat.v1.summary.merge_all.
  • QUEUE_RUNNERS: объекты QueueRunner, используемые для подготовки входных данных для вычислений. Подробнее см. в разделе tf.compat.v1.train.start_queue_runners.
  • MOVING_AVERAGE_VARIABLES: подмножество объектов Variable, для которых также будут храниться скользящие средние. Подробнее см. в разделе tf.compat.v1.moving_average_variables.
  • REGULARIZATION_LOSSES: потери на регуляризацию, собираемые во время построения графа.

Следующие стандартные ключи определены, но их коллекции, в отличие от многих других, не заполняются автоматически:

  • WEIGHTS
  • BIASES
  • ACTIVATIONS
Переменные класса
ACTIVATIONS 'activations'
ASSET_FILEPATHS 'asset_filepaths'
BIASES 'biases'
CONCATENATED_VARIABLES 'concatenated_variables'
COND_CONTEXT 'cond_context'
EVAL_STEP 'eval_step'
GLOBAL_STEP 'global_step'
GLOBAL_VARIABLES 'variables'
INIT_OP 'init_op'
LOCAL_INIT_OP 'local_init_op'
LOCAL_RESOURCES 'local_resources'
LOCAL_VARIABLES 'local_variables'
LOSSES 'losses'
METRIC_VARIABLES 'metric_variables'
MODEL_VARIABLES 'model_variables'
MOVING_AVERAGE_VARIABLES 'moving_average_variables'
QUEUE_RUNNERS 'queue_runners'
READY_FOR_LOCAL_INIT_OP 'ready_for_local_init_op'
READY_OP 'ready_op'
REGULARIZATION_LOSSES 'regularization_losses'
RESOURCES 'resources'
SAVEABLE_OBJECTS 'saveable_objects'
SAVERS 'savers'
SUMMARIES 'summaries'
SUMMARY_OP 'summary_op'
TABLE_INITIALIZERS 'table_initializer'
TRAINABLE_RESOURCE_VARIABLES 'trainable_resource_variables'
TRAINABLE_VARIABLES 'trainable_variables'
TRAIN_OP 'train_op'
UPDATE_OPS 'update_ops'
VARIABLES 'variables'
WEIGHTS 'weights'
WHILE_CONTEXT 'while_context'

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

Spec-Zone.ru

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