Pytorch example for Autoencoder + classifier

View here.