Autograd profiler does not finish

Hi, I need some help as I can’t figure out the issue. If I run my code with cProfile, it works fine. However, if I use the autograd profiler, it never finishes running. Any idea what the issue might be?

As a side note, I have similar issues when I include torch.nn.DataParallel. I can include some code if needed but it is quite long.