mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source 
* Suppress torch.jit.TracerWarning

TracerWarnings can be safely ignored.

* Cleanup
This commit is contained in:
Glenn Jocher 2022-02-01 22:59:26 +01:00 committed by GitHub
parent 77977e0791
commit 842d049e1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
export.py
View File

@ -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__":