speechbrain.utils.profiling 模块

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

作者

  • Titouan Parcollet 2024

摘要

函数

prepare_profiler

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

参考

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

用于创建 PyTorch 分析器以对 speechbrain.core.Brain 实例训练进行基准测试的封装器。详细信息请参见 torch.profiler.profile 文档（下方为简要摘要）。

参数:

  • profile_warmup (int) – 在开始记录之前进行预热的步数。

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

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

返回类型:

profiler