Fused Adam is now monkey patched inside pyprof.init().

NVIDIA · Jun 16, 2020 · 9f04421 · 9f04421
1 parent 6a8f571
commit 9f04421
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/pyprof/examples/imagenet/imagenet.py b/pyprof/examples/imagenet/imagenet.py
@@ -238,7 +238,6 @@ def main():
     args = parseArgs()
 
     pyprof.init()
-    #	pyprof.nvtx.wrap(fused_adam_cuda, 'adam')
 
     N = args.b
     C = 3