Comment out dataloader patching

NVIDIA · Feb 8, 2021 · b67a981 · b67a981
1 parent 708b169
commit b67a981
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()