tf.debugging.experimental.enable_dump_debug_info
Включает выгрузку отладочной информации из программы TensorFlow.
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 | Режим отладки значений тензоров в виде строки. В настоящее время поддерживаются следующие варианты:
|
circular_buffer_size | Размер циклических буферов для событий выполнения. Эти циклические буферы предназначены для снижения накладных расходов на отладку. Они содержат самые последние отладочные события, относящиеся к жадному выполнению операций и tf.function и трассировки значений тензоров, вычисленных внутри tf.function. Они записываются на файловую систему только при вызове соответствующего метода сброса (см. описание возвращаемых значений ниже). Ожидается, что это будет целое число. Если значение <= 0, поведение циклического буфера будет отключено, т. е. события отладки выполнения будут записываться в файловые записи так же, как и события невыполнения, такие как создание операций и снимки файлов исходного кода. |
op_regex | Выгружать данные только из тензоров с типами операций, соответствующими регулярному выражению (через Python's re.match()). "Тип операции" относится к именам операций TensorFlow (например, "MatMul", "LogSoftmax"), которые могут повторяться в функции TensorFlow. Это не относится к именам узлов (например, "dense/MatMul", "dense_1/MatMul_1"), которые уникальны в рамках функции. op_regex="^(MatMul|Relu)$".op_regex="(?!^Relu$)". Этот фильтр работает в логическом операторе AND с tensor_dtypes. |
tensor_dtypes | Выгружать данные только из тензоров указанных типов. Этот необязательный аргумент может иметь любой из следующих форматов: DType объектов или строк, которые можно преобразовать в DType объекты с помощью tf.as_dtype(). Примеры:
DType аргумент и возвращает Python boolean, указывающий, должен ли тип включаться в выгрузку данных. Примеры:
|
| Возвращаемое значение | |
|---|---|
Экземпляр 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