I have got a training data set with dimensions (32, 32, 3, 10000) testing dataset with dimensions (32, 32, 3, 1000). I am trying to build a cnn model to predict the image. I