We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
nn.Transpose3d
Hi, refer to https://github.com/NVIDIA-AI-IOT/torch2trt/blob/master/torch2trt/converters/native_converters.py?plain=1#L439, out_channels = int(weight.shape[0]) should be modified as out_channels = int(weight.shape[1]). ?
out_channels = int(weight.shape[0])
out_channels = int(weight.shape[1])
For transpose3d, weight was defined (in_channels,outchannels, ...) which was exactly opposite of nn.conv3d.
(in_channels,outchannels, ...)
The text was updated successfully, but these errors were encountered:
No branches or pull requests
Hi,
refer to https://github.com/NVIDIA-AI-IOT/torch2trt/blob/master/torch2trt/converters/native_converters.py?plain=1#L439,
out_channels = int(weight.shape[0])
should be modified asout_channels = int(weight.shape[1])
. ?For transpose3d, weight was defined
(in_channels,outchannels, ...)
which was exactly opposite of nn.conv3d.The text was updated successfully, but these errors were encountered: