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