What should be the input size of first linear layer?

1

pytorchhh
pytorchhh888×414 12.3 KB

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

2

Can you try

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

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

4

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