torch_npu.npu.iteration_start
API接口
torch_npu.npu.iteration_start()
功能描述
为了从prof图上区分step信息,进入每个step时打上开始标记。
父主题:
torch_npu.npu profiler