shape issue for y_pred for a custom made loss function about tf-keras

[tilakrayal]

@saluisto,
The error is because you adapt your code from a code with the original input image size 24*24. The tensor shape after two convolution and two max-pooling layers is [-1, 6, 6, 64]. However, as your input image shape is 150*150, the intermediate shape becomes [-1, 38, 38, 64].

The image size and the model's input shape were different. Could you please check your image size again.
Also take a look at this comment for the similar error. Thank you!

[saluisto]

Thank you for your answer. But I am not sure if I understand your answer correctly: In my script I am not using a convolution layer neither a max-pooling layer. I only use dense layer architecture (see below). Input shape should be (-1,100)

Layer (type) Output Shape Param #

dense (Dense) (None, 100) 200

dense_1 (Dense) (None, 200) 20200

dense_2 (Dense) (None, 300) 60300

dense_3 (Dense) (None, 300) 90300

dense_4 (Dense) (None, 200) 60200

dense_5 (Dense) (None, 100) 20100

=================================================================
Total params: 251,300
Trainable params: 251,300
Non-trainable params: 0

[saluisto]

I resolved the issue. Thank you for your help