2eef37ca549b2460ab3c3979161c727a7deb92df,examples/profile.py,,main,#Any#,38

Before Change

def main(parser):
    opt = parser.parse_args()
    with torch.autograd.profiler.profile(
        use_nvprof=opt["use_nvprof"], trace_path=opt["trace_path"]
    ) as prof:
        train(parser)
    print(prof.total_average())

After Change

        pr.enable()
        train(parser)
        pr.disable()
        s = io.StringIO()
        sortby = "cumulative"
        ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
        ps.print_stats()
        print(s.getvalue())
        if opt["debug"]:
            pdb.set_trace()

Instances