-
Hi,
I serialized the model, and then loaded the model in C++ API following https://github.com/NVIDIA-AI-IOT/torch2trt/issues/16
But when I sent a tensor with all ones into the models (pytorch mo…
-
hello!
When executing the following code I get an initialisation error
modelz = torchvision.models.alexnet(pretrained=True).eval().cuda()
x = torch.ones((1,3,224,224)).cuda()
model_trt = torch…
-
Hello. I run the sample under the python2.7 but it doesn't run.
I follow the readme and write the following test.
```
import torch
from torch2trt import torch2trt
from torchvision.models.alexnet …
-
I install torch2trt by --plugins, and I succeed in convert my model to tensorrt model.
And I use torch.save(model_trt.state_dict(), 'test.pth')
then
model_trt = TRTModule()
model_trt.load_…
-
I have installed this repo with plugins, but when I issue `torch2trt.CONVERTERS.keys()`:
![image](https://user-images.githubusercontent.com/12935189/61279780-59f5a480-a7e9-11e9-9244-d6363b4fe9de.png)…
-
Hi, when I tried to run the example code in README, error occurs like this:
```
model_trt = torch2trt(net, [x])
File "build/bdist.linux-x86_64/egg/torch2trt/torch2trt.py", line 233, in torc…
-
Hello, I like this awesome repository and it will be very helpful, but I have some trouble in installing it.
When I tried to install: python setup.py install --plugins
it gived:
![creenshot_2019081…
-
when using trt.Builder(TRT_LOGGER) as builder, an error came out:
```
[TensorRT] ERROR: Cuda initialization failure with error 1. Please check cuda installation: http://docs.nvidia.com/cuda/cuda-in…
-
Hi guys,
I've been working with the repo for a few weeks now and everything went great until I tried to update the repo to get the new MobileNet implementation.
I tried to rebuild the repo after …