Unverified 提交 842d049e authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

* Suppress torch.jit.TracerWarning TracerWarnings can be safely ignored. * Cleanup
上级 77977e07
...@@ -45,6 +45,7 @@ import platform ...@@ -45,6 +45,7 @@ import platform
import subprocess import subprocess
import sys import sys
import time import time
import warnings
from pathlib import Path from pathlib import Path
import torch import torch
...@@ -508,6 +509,8 @@ def parse_opt(): ...@@ -508,6 +509,8 @@ def parse_opt():
def main(opt): def main(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]): for opt.weights in (opt.weights if isinstance(opt.weights, list) else [opt.weights]):
run(**vars(opt)) run(**vars(opt))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论