# ARAE Code Note AutoEncoder Structure (2 Style) ![](https://i.imgur.com/Zdr54KG.png) ### train_ae 1. Reconstruction loss ![](https://i.imgur.com/gW0eUcQ.png) CrossEntropy( [Input0, Input1, ….], [Out0, Out1, ….] ) ### train_classifier 2. Style loss CrossEntropy( Train1 data, 1 ) CrossEntropy( Train0 data, 2) ### GAN related code 3. GAN loss ![](https://i.imgur.com/LEnMijI.png)