augmentation layer computes and stores mean; euclidean loss layer can have coefficient; training script outputs all non-zero losses rather than just the sum (as it used to be)