i am facing the same problem…
hello, try to set trainig image axis=3, it worked for me , hope it will work for u too.
training_images = np.expand_dims(training_images,axis=3)