Description

Missing required arguments in calling torch.nn.modules.conv._QuantConvTransposeNd._output_padding from pytorch_quantization.nn.modules.quant_conv.QuantConvTranspose1d, pytorch_quantization.nn.modules.quant_conv.QuantConvTranspose2d, pytorch_quantization.nn.modules.quant_conv.QuantConvTranspose3d.

_ConvTransposeNd:

    def _output_padding(self, input: Tensor, output_size: Optional[List[int]],
                        stride: List[int], padding: List[int], kernel_size: List[int],
                        num_spatial_dims: int, dilation: Optional[List[int]] = None) -> List[int]:

Whereas called in QuantConvTranspose1d:

        output_padding = self._output_padding(input, output_size, self.stride, self.padding, self.kernel_size)

Environment

TensorRT Version: *

NVIDIA GPU: *

NVIDIA Driver Version: *

CUDA Version: *

CUDNN Version: *

Operating System:

Python Version (if applicable):

Tensorflow Version (if applicable):

PyTorch Version (if applicable):

Baremetal or Container (if so, version):

Relevant Files

Model link:

Steps To Reproduce

Commands or scripts:

Have you tried the latest release?:

Can this model run on other frameworks? For example run ONNX model with ONNXRuntime (polygraphy run <model.onnx> --onnxrt):