Transfer Learning evaluation error

Expected 4-dimensional input for 4-dimensional weight [64, 3, 3, 3], but got 2-dimensional input of size [394, 25088] instead

My data is already in batch of 394x3x224x224, still i am getting this error

Make sure your training, validation and test data is correct.

Here it seems that even the batch loading is wrong. You should have a batch of images, each with a 3 channels.

Check how your data looks before any evaluation. Ideally, you should be able to display a whole batch of images.