Suppress `torch.jit.TracerWarning` on export (#6498)

export.py

@@ -45,6 +45,7 @@ import platform
 import subprocess
 import sys
 import time
+import warnings
 from pathlib import Path
 
 import torch
@@ -508,8 +509,10 @@ def parse_opt():
 
 
 def main(opt):
-    for opt.weights in (opt.weights if isinstance(opt.weights, list) else [opt.weights]):
-        run(**vars(opt))
+    with warnings.catch_warnings():
+        warnings.filterwarnings(action='ignore', category=torch.jit.TracerWarning)  # suppress TracerWarning
+        for opt.weights in (opt.weights if isinstance(opt.weights, list) else [opt.weights]):
+            run(**vars(opt))
 
 
 if __name__ == "__main__":