Keras model building get error Cannot convert ’51’ to a shape

I am making a keras model to classify Human pose. I have taken the code from the link

tensorflow.keras.layers.Flatten() throwing INVALID_ARGUMENT error

I have created this computer vision model using tensorflow.keras with grouped convolutions this way: