What should be the input size of first linear layer?

i am getting error - size mismatch, m1: [10 x 32], m2: [320 x 564] at C:\cb\pytorch_1000000000000\work\aten\src\TH/generic/THTensorMath.cpp:41

Can you try

self.fc1 = nn.Linear(32,564)

thanks for the answer, but i have one doubt since my batch size is 10 then does the images get loaded one by one or simultaneously

The complete input batch will be used and you don’t have to specify the batch size in any layer arguments, such as the number of input features etc.

1 Like