Skip to content

Conversation

geoffrey-g-delhomme
Copy link

Fix #2964

@samurdhikaru
Copy link
Collaborator

/blossom-ci

2 similar comments
@samurdhikaru
Copy link
Collaborator

/blossom-ci

@samurdhikaru
Copy link
Collaborator

/blossom-ci

@rajeevsrao rajeevsrao requested a review from ttyio November 29, 2023 14:25
@ttyio
Copy link
Collaborator

ttyio commented Nov 29, 2023

thanks for the contributing, the latest code is in branch release/9.1, and for compatibility, we inspect the signature first to see if those spatial dims and dilations are needed. see:

https://github.com/NVIDIA/TensorRT/blob/release/9.1/tools/pytorch-quantization/pytorch_quantization/nn/modules/quant_conv.py#L278

@yjszyd0813
Copy link

When will the pytorch quantification package be updated synchronously for this issue? It's too troublesome to modify the source code every time.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Missing required positional arguments calling _output_padding in _ConvTransposeNd from torch.
4 participants