Correction: the auto encoder is pre-trained :)