torch2trt
torch2trt copied to clipboard
An easy to use PyTorch to TensorRT converter
Hello! My name is McKenna Dallmeyer and I would like to submit torch2trt to OSS-Fuzz. If you are not familiar with the project, OSS-Fuzz is Google's platform for continuous fuzzing...
This update addresses the handling of None values in the output size when converting adaptive average pooling operations to TensorRT. Previously, the conversion process did not account for cases where...
I use tons of torch.Tensor.to methods in my model for moving tensor between cpu/cuda, but the method is not supported in torch2trt currently. Will it ever be supported in the...
here is a simple reproduce code ``` import os import time import torch import torchvision from torch2trt.torch2trt import * import numpy as np class Sparse(torch.nn.Module): def __init__(self, embedding_size): super().__init__() self._embedding_size...
Hello, I would like to suggest adding a ProfilingVerbosity option to the configuration when building the TensorRT engine in torch2trt. This option would allow users to set the verbosity level...
Hello, and thank you for your outstanding project. I encountered an error when converting a structure containing ConvTranspose2d using torch2trt. Here is the code and the error. ```python import torch...
DEPRECATION: Loading egg at e:\miniconda\lib\site-packages\torch2trt-0.5.0-py3.11.egg is deprecated.
DEPRECATION: Loading egg at e:\miniconda\lib\site-packages\torch2trt-0.5.0-py3.11.egg is deprecated. pip 24.3 will enforce this behaviour change. A possible replacement is to use pip for package installation.
i use jetson orin nano. i tried to install this package in jetson orin nano for export pytorch model to tensorRT. i tried `python setup.py install` but it's not work...
Hello, I am beginner about deep learning. I am using PatchCore anomaly detection algorithm for detecting defects. I want to accelerate PatchCore with TensorRT. But I met some failure. First,...