search

IntelLabs / bayesian-torch

A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch
BSD 3-Clause "New" or "Revised" License
544 stars 73 forks source link

problem in conv_transpose2d #34

Closed amirshamaei closed 10 months ago

amirshamaei commented 11 months ago

Hi, there is a problem in conv_transpose2d

 out = F.conv_transpose2d(input, weight, bias, self.stride,
                                 self.padding, self.output_padding,
                                 self.dilation, self.groups)

According to pytorch doc, it should be first groups, and then dilation

ranganathkrishnan commented 10 months ago

Thanks @amirshamaei for identifying the problem. Fixed with commit 97ba16ad044022035ba22b17aa279f2d389129eb.