It looks like the in_features of your linear layer are set to 1024.
Could this be the case?
If so, you should define them as in_features=2420, since that’s the number of features from your conv layers.