I try to train my model, which is composed of 2 deep neural networks, with different input shapes.
# import from the same image dataset train_dataset = tf.ke