The code for the vaeac is coming from : https://github.com/tigvarts/vaeac