You could use the PyTorch profiler or Nsight Systems to get a timeline view.
If you want to profile specific operations, you could use torch.utils.benchmark instead.
1 Like
You could use the PyTorch profiler or Nsight Systems to get a timeline view.
If you want to profile specific operations, you could use torch.utils.benchmark instead.