Imported from https://github.com/mseitzer/neurips2019-disentanglement-challenge