IntelLabs / bayesian-torch

A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

problem in conv_transpose2d

amirshamaei opened this issue · comments

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

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