bayesian-torch icon indicating copy to clipboard operation
bayesian-torch copied to clipboard

conv_transpose2d parameter order problem

Open pierreosselin opened this issue 3 years ago • 1 comments
trafficstars

https://github.com/IntelLabs/bayesian-torch/blob/f6f516e9b3721466aa0036c735475a421cc3ce80/bayesian_torch/layers/variational_layers/conv_variational.py#L784-L786

image

When I try to convert a deterministic autoencoder into a bayesian one with a ConvTranspose2d layer I get the error "Exception has occurred: TypeError conv_transpose2d(): argument 'groups' (position 7) must be int, not tuple" which I suspect comes from self.dilation and self.group which are swaped.

pierreosselin avatar May 17 '22 15:05 pierreosselin

@pierreosselin can you check how you are getting model output in your code? If return_kl flag is True (by default), the forward function returns a tuple (output and kl) - https://github.com/IntelLabs/bayesian-torch/blob/f6f516e9b3721466aa0036c735475a421cc3ce80/bayesian_torch/layers/variational_layers/conv_variational.py#L793

ranganathkrishnan avatar Jul 22 '22 20:07 ranganathkrishnan

closing the issue due to no response from the user.

ranganathkrishnan avatar Sep 15 '22 15:09 ranganathkrishnan

I am having the same issue and return_kl flag is False

SoyedTuhinAhmed avatar Oct 31 '22 22:10 SoyedTuhinAhmed

I am having the same Issue, pls suggest what can be wrong

rameshamurtaza avatar Mar 03 '23 11:03 rameshamurtaza