Update infer.py

RandyWangRaining · Dec 16, 2024 · 6844199 · 6844199
1 parent b9d145d
commit 6844199
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/infer.py b/tools/infer.py
@@ -33,7 +33,7 @@ def parse_config():
     args.run_mode = 'infer'
     return args, cfgs
 
-
+@torch.no_grad()
 def main():
     args, cfgs = parse_config()
     if args.dist_mode: