Skip to content

Allow onnx kwargs#717

Open
lazToum wants to merge 6 commits into
NVIDIA-AI-IOT:masterfrom
lazToum:master
Open

Allow onnx kwargs#717
lazToum wants to merge 6 commits into
NVIDIA-AI-IOT:masterfrom
lazToum:master

Conversation

@lazToum

@lazToum lazToum commented Apr 6, 2022

Copy link
Copy Markdown

Example usage:

model_trt = torch2trt(
    model,
    [inputs],
    max_batch_size=max_batch_size,
    use_onnx=True,
    onnx_kwargs={'opset_version': 13, 'do_constant_folding': True}
)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant