TylerYep / torchinfo

View model summaries in PyTorch!

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Bug parsing bias argument

amandalucasp opened this issue · comments

Describe the bug

Get the following message:

TypeError: conv2d() received an invalid combination of arguments - got (Tensor, Parameter, NoneType, tuple, tuple, tuple, int), but expected one of:  * (Tensor input, Tensor weight, Tensor bias, tuple of ints stride, tuple of ints padding, tuple of ints dilation, int groups)

To Reproduce

I think it might be related to the way I coded the first layer of the net? Not sure.

self.conv1 = nn.Conv2d(self.in_channels, self.inplanes, kernel_size=7, stride=strides[0], padding=3, bias=False)