Hi,

The warning doesn’t appear when I use 1568 as input for the flatten layer. I just don’t understand why.

What other shape do you use for the input of your flatten layer? The output of your last conv is of shape [B, 8, 14,14] -> flattened [B, 1568], so anything else would raise a size mismatch error.