-
dynamic batch is not possible even though I defined min and max shapes or defined max_batch_size.
```
x = torch.rand([1,1,5,5,2000]).cuda().half()
model = model.half()
model = torch2tr…
-
Hi,
I'm trying to use torch2trt for nn.GroupNorm layer. When I run my toy model with torch2trt, it outputs the following error:
```
import torch
import torch.nn as nn
from torch2trt import t…
-
Receiving a compile error when attempting ```python setup.py install --plugins``` Will track down and add further detail but wanted to add the issue here.
```
C:\Users\vroom\Documents\Bitbucket\…
-
Hi, I now want to convert my model which contains DCNv2 to tensorRT. But DCNv2 is not supported in tensorRT.
So I find the DCNv2 plugin for tensorRT (https://github.com/eweill-nv/dcnv2_trt) but thi…
-
**Describe the bug**
I'm using the ```mmdet2trt ../mmdetection/configs/detr/detr_r50_8x2_150e_coco.py ./detr_r50_8x2_150e_coco_20201130_194835-2c4b8974.pth detr.trt```to convert DETR to TensorRT…
-
**Problem**
The `torch2trt` conversion for `torch.Tensor.__getitem__` fails when indexing/slicing for single element; eg.
```
tensor = torch.rand(2, 3)
tensor[0]
```
The issue is in `torch2trt/c…
-
### Description:
I encountered an error while using torch2trt for converting a PyTorch model with Transformer operator. The error message is as follows:
```
Traceback (most recent call last):
..…
Thrsu updated
9 months ago
-
tensorrt 6.0.18 ubuntu 18.04 python 3.7
I want to convert model ( https://github.com/shariqfarooq123/AdaBins)
import torch
from torch2trt import torch2trt
from models import UnetAdaptiveBins
imp…
-
I'm implementing my own converter and I'm trying to figure out the general framework for testing converters in this repository.
For example, I'm trying to test `Conv2d` converter via executing the co…
-
It worked well without plugins, but when I installed with plugins, It reported an error:
`python setup.py install --plugins
running install
Building in plugin support
[2/3] g++ -c -fPIC torch2tr…