Getting an error while implementing the global average pooling layer

#1

532bf4a3-c792-4117-a41e-aacd8a5d9456.png2876×1384

#2

hey @udayjain00 ,
check this

#3

thanks for helping with the code

#4

pls share the code for fitting and predicting outputs

#5

for that i would suggest you to kindly try yourself once.
Handle the errors and if still you are stuck then let me know.