So I am trying to write the code for Cat/Dog Pet classifier using TensorFlow 2. I have done all the layers and when I attempt to fit the model I get the error \'InvalidArgum