I\'m training a convolutional neural network (more specifically, GAN) with tensorflow 2.4 and python 3.7 on GPU. It trains successfully for some epoches but then runs into l