更新时间:2024-12-30 GMT+08:00
分享

Profiling数据采集

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

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

相关文档