From 9f044218ec713b306d02186981e38525e270e2ba Mon Sep 17 00:00:00 2001
From: Aditya Agrawal <aditya.iitb@gmail.com>
Date: Fri, 12 Jun 2020 04:13:31 +0000
Subject: [PATCH] Fused Adam is now monkey patched inside pyprof.init().

---
 pyprof/examples/imagenet/imagenet.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/pyprof/examples/imagenet/imagenet.py b/pyprof/examples/imagenet/imagenet.py
index d7e2fbf..6c3e8b0 100644
--- 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