apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

lint(api): suppress some known warnings during model conversion (#75)

This commit is contained in:
Sean Sube 2023-01-22 15:45:09 -06:00
parent 2921eba1f6
commit 1b2618c0f7
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/convert.py
View File

@ -11,6 +11,11 @@ from torch.onnx import export
from typing import Dict, List, Tuple, Union from typing import Dict, List, Tuple, Union
import torch import torch
import warnings
# suppress common but harmless warnings, https://github.com/ssube/onnx-web/issues/75
warnings.filterwarnings('ignore', '.*The shape inference of prim::Constant type is missing.*')
warnings.filterwarnings('ignore', '.*Only steps=1 can be constant folded.*')
Models = Dict[str, List[Tuple[str, str, Union[int, None]]]] Models = Dict[str, List[Tuple[str, str, Union[int, None]]]]