-
```
[07/22/2024-10:21:17] [TRT] [I] [MemUsageChange] Init CUDA: CPU +14, GPU +0, now: CPU 24, GPU 17382 (MiB)
[07/22/2024-10:21:19] [TRT] [I] [MemUsageChange] Init builder kernel library: CPU +2052,…
-
The following code:
```python
import torch
from torch2trt import torch2trt
import timm
vit_model = timm.create_model(
model_name="vit_base_patch16_224_dino",
pretrained=True,
)
…
-
when use torch2trt convert the torch.eq, error occurs.
`mm = torch.eq(mm, 0.)`
mm is tensor and mm.shape = [3136, 1, 3, 3]
File "/media/cfs/torch2trt-master/examples/inpainting/model.py", line 32…
whcjb updated
4 weeks ago
-
Hi Guys,
I tried to run this repo on Jetson nano, and I did the same steps in the instruction.
Somehow it failed to convert model to TRT.
It seems the dimension is not correct?
```
>>> impor…
-
Hi,
When I try to install torch2trt, I faced with a number of error and the installation will fail at the end.
it is a long list of errors so I cannot paste the entire message here but I picked up…
AffDk updated
3 years ago
-
Hi,
I've been trying to convert a yolov7 model to tensorrt format but I get a bunch of the following printouts"
```
[08/30/2022-15:24:57] [TRT] [E] 3: [builderConfig.cpp::canRunOnDLA::382] Erro…
-
Hi,The following error occurs when performing the conversion of a torch model to a tensorRT model:TypeError: forward() missing 1 required positional argument: 'multimask_output'.
But I trained th…
-
你的yolo.py需要torch2trt,但是pip install torch2trt 发现ERROR: No matching distribution found for torch2trt
想问下你的pip install torch2trt 在哪下载
-
### Description:
I'm experiencing a discrepancy between the inference results of PyTorch model and the TensorRT model obtained by converting it using the torch2trt tool.
### Reproduce
1. BatchN…
-
Hello,
First off, I would like to thank you for your work and willingness to make it opensource. I was following the instructions in the README of the torch2trt folder, and the current visualization …