Profiling数据采集
在train.py的main()函数Step迭代处添加配置,添加位置如下图所示:
此处需要注意的是prof.step()需要加到dataloder迭代循环的内部以保证采集单个Step迭代的Profiling数据。
更多信息,请参见Ascend PyTorch Profiler接口采集。
检测到您已登录华为云国际站账号,为了您更好的体验,建议您访问国际站服务网站 https://www.huaweicloud.com/intl/zh-cn
不再显示此消息
在train.py的main()函数Step迭代处添加配置,添加位置如下图所示:
此处需要注意的是prof.step()需要加到dataloder迭代循环的内部以保证采集单个Step迭代的Profiling数据。
更多信息,请参见Ascend PyTorch Profiler接口采集。