|
|
@ -1,9 +1,8 @@ |
|
|
|
|
|
|
|
from .engine import EngineBuilder, TRTModule, TRTProfilerV0, TRTProfilerV1 # isort:skip # noqa: E501 |
|
|
|
import warnings |
|
|
|
import warnings |
|
|
|
|
|
|
|
|
|
|
|
import torch |
|
|
|
import torch |
|
|
|
|
|
|
|
|
|
|
|
from .engine import EngineBuilder, TRTModule, TRTProfilerV0, TRTProfilerV1 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
warnings.filterwarnings(action='ignore', category=torch.jit.TracerWarning) |
|
|
|
warnings.filterwarnings(action='ignore', category=torch.jit.TracerWarning) |
|
|
|
warnings.filterwarnings(action='ignore', category=torch.jit.ScriptWarning) |
|
|
|
warnings.filterwarnings(action='ignore', category=torch.jit.ScriptWarning) |
|
|
|
warnings.filterwarnings(action='ignore', category=UserWarning) |
|
|
|
warnings.filterwarnings(action='ignore', category=UserWarning) |
|
|
|