- class ConfigOnnxToTRT(input_model=None, output_model=None, batches=<factory>, seq_length=None, max_workspace_size=16000)[source]
Bases:
morpheus.config.ConfigBase
Configuration class for the OnnxToTRT migration process.
- Parameters
- input_modelstr
- output_modelstr
- batchestyping.List[typing.Tuple[int, int]]
- seq_lengthint
- max_workspace_sizeint
Input ONNX model file.
Output TensorRT model file.
Batches of input data.
Sequence length.
Maximum workspace size, default 16000 MB.
- Attributes
- input_modelstr
- output_modelstr
- seq_lengthint
Input ONNX model file.
Output TensorRT model file.
Sequence length.