Does anyone know how to fix a data leak with tensorflow-gpu 2.3.0 while training a model using a data pipeline for tfrecords?

Question

I am trying to train a convolutional autoencoder on tfrecords I generated from 4K images. I have been able to successfully set up an input pipeline using TF Dataset API and