From b67a981f0d865fa4359a64097bd5487785dc89ff Mon Sep 17 00:00:00 2001 From: Elias Bermudez Date: Mon, 8 Feb 2021 13:00:32 -0800 Subject: [PATCH] Comment out dataloader patching --- pyprof/nvtx/nvmarker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()