You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11 lines
460 B
11 lines
460 B
import warnings |
|
|
|
import torch |
|
|
|
from .engine import EngineBuilder, TRTModule, TRTProfilerV0, TRTProfilerV1 |
|
|
|
warnings.filterwarnings(action='ignore', category=torch.jit.TracerWarning) |
|
warnings.filterwarnings(action='ignore', category=torch.jit.ScriptWarning) |
|
warnings.filterwarnings(action='ignore', category=UserWarning) |
|
warnings.filterwarnings(action='ignore', category=FutureWarning) |
|
__all__ = ['EngineBuilder', 'TRTModule', 'TRTProfilerV0', 'TRTProfilerV1']
|
|
|