-
## Environment
Ubuntu 20.04
Python 3.8.8
torch 1.8.1
torch2trt 0.2.0
cuda 11.1
cudnn 8.1.1
TensorRT 7.2.2
## Issue
I'm trying to convert YOLOR(https://github.com/WongKinYiu/yolor) implem…
-
Hi, I am testing onnx-tensorrt backend, block at the first step:
when I ran my test onnx model in tensorrt backend, I got the warns and errors as below:
1.
warn: Your ONNX model has been generate…
-
I'm trying to use this: https://github.com/NVIDIA-AI-IOT/torch2trt
To convert YOLACT to TensorRT for performance reasons. The conversion doesn't seem to be compatible.
Has anyone managed to make t…
-
I get error `AttributeError: 'Tensor' object has no attribute '_trt'` when I meet MaxPool2d
/usr/local/lib/python3.6/dist-packages/torch2trt-0.0.0-py3.6.egg/torch2trt/torch2trt.py in wrapper(*ar…
-
just stumble upon your repo when trying to use real-esrgan with tensorrt, i think the instructions can be a bit better, especially windows and the `.bat` file
nvidia docs are very long and complica…
-
Currently, the `torch.squeeze` converter removes dimensions without checking if it's valid to do so; ie. dimensions should only be removed if they are singular (ie. 1). Otherwise, removing dimensions …
-
Hi, I was wondering if anyone knew how can I use my torch2trt engine on images and get back the tensor with detections in it ? I know that i need to preprocess my image to send it to my model but afte…
-
Hi, I am trying to apply torch2trt on [FairMot model](https://github.com/ifzhang/FairMOT). It has an external library DCNv2.
1)With option fp16_mode=True, DCNv2 cannot be converted correctly and me…
-
Hi. I'm trying using torch2trt_dynamic to convert models from torch to trt. I got an unexpected issue.
To reproduce it I use lenet5 like neural network.
```
from torch2trt_dynamic import torch2trt_…
-
I tried to install torch2trt with plugins with the command **' python setup.py install --plugins'** on Windows 10 after changing the lines among from No.6 to No.10 in **setup.py** as:
###
def tr…