Profiling数据采集

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

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

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