torch.autograd.profiler.load_nvprof

torch.autograd.profiler.load_nvprof(path)

Open an nvprof trace file and parses autograd annotations.

参数

path (str) – path to nvprof trace