I am trying to train a convolutional autoencoder using tensorflow.keras. Here, I have a Model Class API (I think that is what it\'s called) that works using a single GPU (no