Skip to content

fix: use ONNX backend for torch2trt conversion in notebooks - #188

Open
Mrudula-itsjuzme wants to merge 1 commit into
NVIDIA-AI-IOT:masterfrom
Mrudula-itsjuzme:fix-torch2trt-onnx
Open

fix: use ONNX backend for torch2trt conversion in notebooks#188
Mrudula-itsjuzme wants to merge 1 commit into
NVIDIA-AI-IOT:masterfrom
Mrudula-itsjuzme:fix-torch2trt-onnx

Conversation

@Mrudula-itsjuzme

Copy link
Copy Markdown

Resolves #178 by adding use_onnx=True to the torch2trt.torch2trt calls in the live_demo.ipynb and eval.ipynb notebooks. This prevents the __len__() should return >= 0 ValueError associated with the native torch2trt converter, ensuring the notebooks run out of the box on modern environments.

Signed-off-by: Mrudula-itsjuzme <mrudulasankar2007@gmail.com>
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.

Error while running live_demo.ipynb, __len__() should return >= 0

1 participant