Wrong shape at fully connected layer: mat1 and mat2 shapes cannot be multiplied
I have the following model. It is training well. The shapes of my splits are:
I have the following model. It is training well. The shapes of my splits are: