-
I try to convert TridentNet to engine, but I have an error:
load checkpoint from local path: /home/ariel/mmdetection2/tridentnet/epoch_33.pth
can't find wrap module for type:, use instead.
can't…
-
I just saw that v0.4.0 was released a few hours ago. Nice!
Looking through the changes, I noticed the following code snippet:
```
# handle inputs as dataset of list of tensors
if issubc…
-
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_…
-
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…
-
Hi All,
Thank you for the amazing work! Can this model be exported to TensorRT for inference?
Thanks
-
I'm converting a PyTorch model with MobileNetV2 backbone, However I get this error:
`AttributeError: 'tensorrt_bindings.tensorrt.INetworkDefinition' object has no attribute 'add_convolution'. Did y…
-
I trained yolox-m model and convert it to TensorRT-int8 model with demo/trt.py.
Difference from original trt.py is torch2trt's args , `float16_mode=false` and `int8_mode=True`, and calibration dat…
-
HI, thanks for your repo, the performance seems to be amazing ! I'm trying to run it on my Jetson Xavier but I'm not able to find out why I have this issue, any guess ? Thanks !
`Start test speed
F…
-
hi, when I was running this code:
import torch
import os
from torch2trt import torch2trt
from torchvision.models.resnet import resnet18
os.environ["CUDA_VISIBLE_DEVICES"]="2"
model = resnet1…
-
For the following code:
```
import torch
from torch import nn
from torch2trt import torch2trt
class Example(nn.Module):
def __init__(self, stride=8):
super(Example, self).__in…