clementchadebec/

reproduced_wrapped_poincare_vae

This model was trained with pythae. It can be downloaded or reloaded using the method load_from_hf_hub

This trained model reproduces the results of the official implementation of [1].

[1] Mathieu, E., Le Lan, C., Maddison, C. J., Tomioka, R., & Teh, Y. W. (2019). Continuous hierarchical representations with poincaré variational auto-encoders. Advances in neural information processing systems, 32.