Spec-Zone.ru › TensorFlow

tf.debugging.experimental.enable_dump_debug_info

Включает выгрузку отладочной информации из программы TensorFlow.

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

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

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

tf.compat.v1.debugging.experimental.enable_dump_debug_info

tf.debugging.experimental.enable_dump_debug_info(
    dump_root,
    tensor_debug_mode=DEFAULT_TENSOR_DEBUG_MODE,
    circular_buffer_size=1000,
    op_regex=None,
    tensor_dtypes=None
)

Отладочная информация выгружается в каталог на файловой системе, указанный как dump_root.

Выгруженная отладочная информация может быть обработана пользовательскими интерфейсами отладчика.

Файлы в каталоге выгрузки содержат следующую информацию:

  • Конструкции функций TensorFlow (например, компиляция функций Python, снабжённых аннотацией @tf.function), типы операций, имена (если доступны), контексты, входные и выходные тензоры и соответствующие трассировки стека.
  • Выполнение операций (ops) и функций TensorFlow, а также их трассировки стека, типы операций, имена (если доступны) и контексты. Кроме того, в зависимости от значения аргумента tensor_debug_mode (см. раздел Аргументы ниже), будут выгружаться значения выходных тензоров или более краткие сводки значений тензоров.
  • Снимок файлов исходного кода Python, участвующих в выполнении программы TensorFlow.

После включения выгрузку можно отключить с помощью соответствующего метода disable_dump_debug_info() в том же пространстве имён Python. Вызов этого метода более одного раза с тем же dump_root является идемпотентным. Вызов этого метода более одного раза с различными tensor_debug_mode приводит к ValueError. Вызов этого метода более одного раза с различными circular_buffer_size приводит к ValueError. Вызов этого метода с другим dump_root отменяет ранее включённую dump_root.

Пример использования:

tf.debugging.experimental.enable_dump_debug_info('/tmp/my-tfdbg-dumps')

# Code to build, train and run your TensorFlow model...
Примечание: Если ваш код выполняется на TPUs, убедитесь, что вы вызываете tf.config.set_soft_device_placement(True) перед вызовом tf.debugging.experimental.enable_dump_debug_info(), так как этот API использует автоматическую внешнюю компиляцию на TPUs. Например:
tf.config.set_soft_device_placement(True)
tf.debugging.experimental.enable_dump_debug_info(
    logdir, tensor_debug_mode="FULL_HEALTH")

resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='')
strategy = tf.distribute.TPUStrategy(resolver)
with strategy.scope():
  # ...
Аргументы
dump_root Путь к каталогу, в который будет записываться информация о выгрузке.
tensor_debug_mode Режим отладки значений тензоров в виде строки. В настоящее время поддерживаются следующие варианты:
  • "NO_TENSOR": (По умолчанию) Только отслеживает выходные тензоры всех выполненных операций (включая те, которые выполняются жадно на уровне Python или как часть графа TensorFlow) и функций, не извлекая никакой информации из значений тензоров.
  • "CURT_HEALTH": Для каждого тензора с плавающей точкой (например, тензоры типов float32, float64 и bfloat16) извлекается двоичный бит, указывающий, содержит ли он -бесконечность, +бесконечность или NaN.
  • "CONCISE_HEALTH": Для каждого тензора с плавающей точкой извлекается общее количество элементов и количество элементов -бесконечность, +бесконечность и NaN.
  • "FULL_HEALTH": Для каждого тензора с плавающей точкой извлекаются тип, ранг (количество измерений), общее количество элементов и количество элементов -бесконечность, +бесконечность и NaN.
  • "SHAPE": Для каждого тензора (независимо от типа) извлекаются его тип, ранг, общее количество элементов и форма.
circular_buffer_size Размер циклических буферов для событий выполнения. Эти циклические буферы предназначены для снижения накладных расходов на отладку. Они содержат самые последние отладочные события, относящиеся к жадному выполнению операций и tf.function и трассировки значений тензоров, вычисленных внутри tf.function. Они записываются на файловую систему только при вызове соответствующего метода сброса (см. описание возвращаемых значений ниже). Ожидается, что это будет целое число. Если значение <= 0, поведение циклического буфера будет отключено, т. е. события отладки выполнения будут записываться в файловые записи так же, как и события невыполнения, такие как создание операций и снимки файлов исходного кода.
op_regex Выгружать данные только из тензоров с типами операций, соответствующими регулярному выражению (через Python's re.match()). "Тип операции" относится к именам операций TensorFlow (например, "MatMul", "LogSoftmax"), которые могут повторяться в функции TensorFlow. Это не относится к именам узлов (например, "dense/MatMul", "dense_1/MatMul_1"), которые уникальны в рамках функции.
  • Пример 1: Выгружать данные тензоров только для операций MatMul и Relu op_regex="^(MatMul|Relu)$".
  • Пример 2: Выгружать тензоры из всех операций, кроме Relu: op_regex="(?!^Relu$)". Этот фильтр работает в логическом операторе AND с tensor_dtypes.
  • tensor_dtypes Выгружать данные только из тензоров указанных типов. Этот необязательный аргумент может иметь любой из следующих форматов:
  • список или кортеж DType объектов или строк, которые можно преобразовать в DType объекты с помощью tf.as_dtype(). Примеры:
    • tensor_dtype=[tf.float32, tf.float64],
    • tensor_dtype=["float32", "float64"],
    • tensor_dtypes=(tf.int32, tf.bool),
    • tensor_dtypes=("int32", "bool")
  • функция, которая принимает один DType аргумент и возвращает Python boolean, указывающий, должен ли тип включаться в выгрузку данных. Примеры:
    • tensor_dtype=lambda dtype: dtype.is_integer. Этот фильтр работает в логическом операторе AND с op_regex.
  • Возвращаемое значение
    Экземпляр DebugEventsWriter, используемый функцией выгрузки. Вызывающий может использовать его методы сброса, включая FlushNonExecutionFiles() и FlushExecutionFiles().

    © 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/debugging/experimental/enable_dump_debug_info

    Spec-Zone.ru

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