My model summary is:
Layer (type) Output Shape Param #
=================================================================
conv2d_1 (Conv2D) (None, 62, 62, 32) 896
_________________________________________________________________
max_pooling2d_1 (MaxPooling2 (None, 31, 31, 32) 0
_________________________________________________________________
conv2d_2 (Conv2D) (None, 29, 29, 32) 9248
_________________________________________________________________
max_pooling2d_2 (MaxPooling2 (None, 14, 14, 32) 0
_________________________________________________________________
flatten_1 (Flatten) (None, 6272) 0
_________________________________________________________________
dense_1 (Dense) (None, 128) 802944
_________________________________________________________________
dense_2 (Dense) (None, 4) 516
While I am re-training this model using the below function:
Im facing this error:
ValueError: Input 0 is incompatible with layer flatten_1: expected min_ndim=3, found ndim=2