You would have to increase the input features of the first linear layer, since the spatial size of the incoming activation would be larger, if you haven’t used adaptive pooling layers.
This would also explain the error message, which points towards a mismatch of the shapes in a linear layer.
I’m not sure, what kind of definitions you are using for the layers and cannot see where the input and output features are defined.