speechbrain.utils.profiling 模块

用于处理PyTorch分析和基准测试的封装器。

Author:

  • Titouan Parcollet 2024

摘要

函数：

prepare_profiler

用于创建 PyTorch 分析器的包装器，以对 speechbrain.core.Brain 实例的训练进行基准测试。

参考

speechbrain.utils.profiling.prepare_profiler(profile_warmup=5, profile_steps=5, logdir='tensorboard_logs')[source]

用于创建 PyTorch 分析器的包装器，以对 speechbrain.core.Brain 实例的训练进行基准测试。 有关详细信息，请参阅 torch.profiler.profile 文档（简要总结如下）。

Parameters:

  • profile_warmup (int) – 在开始记录之前的热身步骤数。

  • profile_steps (int) – 热身后的记录步数。

  • logdir (str) – 日志输出文件夹的路径。

Return type:

性能分析器