I want to use the basic UNET model to reconstruct noisy 512×512 images towards a noiseless image. The TensorFlow implementation of the model is here below. The training gets stuck on the same loss and val_loss for each epoch for at least 200 epochs straight on several learning rates.