Hi all,
I have the following model:
class ModelExample(torch.nn.Module):
def __init__(self):
    super(ModelExample, self).__init__()
    
    self.linear = torch.nn.Linear(100, 1024*4*4)
    
    self.conv1 = nn.Sequential(
                               nn.ConvTranspose2d(in_channels=1024, out_channels=128, kernel_size=4, stride=2, padding=1, bias=False)
                               ,nn.BatchNorm2d(128)
                               ,nn.ReLU(inplace=True)
                              )
                              
    self.conv2 = nn.Sequential(
                               nn.ConvTranspose2d(in_channels=128, out_channels=3, kernel_size=4, stride=2, padding=1, bias=False)
                              )
    self.out = torch.nn.Tanh()

def forward(self, x):
    
    x = self.linear(x)
    x = x.view(x.shape[0], 1024, 4, 4)
    
    x = self.conv1(x)
    x = self.conv2(x)

    x = self.out(x)

    return x

I tried to make the following code work:
model = ModelExample()
summary(model, input_size = (100,64,64), device = ‘cpu’)
But it gives me the following error
RuntimeError: size mismatch, m1: [12800 x 64], m2: [100 x 16384] at …\aten\src\TH/generic/THTensorMath.cpp:41
Can someone please tell me where the error is and how to solve it?
Thanks for your help.