Experiments on more datasets?

[noahcao]

Thanks for the excellent!

I'm curious if you have tried this model on other popular datasets to study disentanglement, e.g., dSprites, SmallNORB, Cars3D, Shapes3D as benchmarked in disentanglement_lib.

I tried to simply migrate your implementation to SmallNORB and dSprites but the training collapses.

It would be wonderful if you could support these benchmarks and related metrics, e.g., BetaVAE score, FactorVAE score, MIG, DCI-disentanglement, SAP. That would provide a nice platform for following works.

Thank you in advance!