1 | torch_npu.profiler.dynamic_profile.step() |
dynamic_profile动态采集划分step。
1 2 3 4 5 6 7 8 9 | # 加载dynamic_profile模块 from torch_npu.profiler import dynamic_profile as dp # 设置Profiling配置文件的路径 dp.init("profiler_config_path") … for step in steps: train_one_step() # 划分step dp.step() |