Output the UNet images are (420, 420). ---depending on different params though
Output the UNet images are (420, 420). ---depending on different params though