更新时间:2024-12-30 GMT+08:00
Profiling数据采集
在train.py的main()函数Step迭代处添加配置,添加位置如下图所示:
此处需要注意的是prof.step()需要加到dataloder迭代循环的内部以保证采集单个Step迭代的Profiling数据。
父主题: 性能调优