diffusion_galaxies

create training script for lensed galaxies

set mse in evaluate, changed, removed custom loader 16 bit, as it is problem specific