morpheus.config.ConfigOnnxToTRT
- 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_model
- output_model
- batches
- seq_length
- max_workspace_size
Input ONNX model file.
Output TensorRT model file.
Batches of input data.
Sequence length.
Maximum workspace size, default 16000 MB.
- Attributes
- input_model
- output_model
- seq_length
Input ONNX model file.
Output TensorRT model file.
Sequence length.