diff --git a/pyprof/nvtx/nvmarker.py b/pyprof/nvtx/nvmarker.py index 10f4e27..2ff739d 100644 --- a/pyprof/nvtx/nvmarker.py +++ b/pyprof/nvtx/nvmarker.py @@ -401,7 +401,7 @@ def init(*args, **kwargs): print("Initializing NVTX monkey patches") patch_apex() - patch_dataloader() + # patch_dataloader() patch_torch_classes() patch_torch_nn_forward_functions()