Hi,
I have an image dataset, each image size of  28 * 28 * 3. I am passing it through two Conv2d/MaxPool2d layer as follows
    self.nn = nn.Sequential(
        nn.Conv2d(in_channels=3, out_channels=32, kernel_size=3, padding=1),#28*28*32
        nn.BatchNorm2d(32),
        nn.ReLU(),
        nn.MaxPool2d(kernel_size=2, stride=2), #14*14*32

        nn.Conv2d(in_channels=32, out_channels=64, kernel_size=3), #12*12*64
        nn.BatchNorm2d(64),
        nn.ReLU(),
        nn.MaxPool2d(2), #10*10*64

        nn.Flatten(),
        nn.Linear(in_features=???) 

I am stuck at what value I need to pass the in_features. If I pass 6400, I get dimension error.
RuntimeError: mat1 dim 1 must match mat2 dim 0
Can someone please help me understand where I am wrong?
Thank you!