I decided to switch to using a tf.Data input for my CNN model and now I am getting the following error message
ValueError: Dimension 0 in both shapes must be equa