Hi, I am learning to train a basic nn model for image classification, the error happened when I was trying to feed in image data into the model. I understand that I should input correct size of image data. My image data is 128*256 with 3 channels. What I don’t understand is where does the number 113216 come from? I checked all related parameters or image meta data, but didn’t find a clue. Here is my code:

image854×806 10.7 KB

Thanks for your help!