Example config for training VAE

[tonyf]

It seems like the repo is missing the yaml for training the autoencoder though the encoder training code is provided?

[coryMosaicML]

Good catch! We will add one soon.

[miquel-espinosa]

Would be great to have instructions for training VAE from scratch on custom dataset

[Atharva-Phatak]

@coryMosaicML any updates on config ?

[coryMosaicML]

Added some example configs to the current PR, you can see them here(for training with our AE implementation) and here(for training an autoencoder from diffusers). Configs are still a bit experimental, so params are likely not optimal.