更新时间:2025-07-29 GMT+08:00
分享

Profiling数据采集

在train.py的main()函数Step迭代处添加配置,添加位置如下图所示:

此处需要注意的是prof.step()需要加到dataloder迭代循环的内部以保证采集单个Step迭代的Profiling数据。

更多信息,请参见Ascend PyTorch Profiler接口采集

相关文档