Training: CelebA normalizing flow =================================================