diffusion icon indicating copy to clipboard operation
diffusion copied to clipboard
verified publisher icon mosaicml

Example config for training VAE

Open tonyf opened this issue 2 years ago • 4 comments

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

tonyf avatar Jan 04 '24 00:01 tonyf

Good catch! We will add one soon.

coryMosaicML avatar Jan 12 '24 03:01 coryMosaicML

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

miquel-espinosa avatar Jan 19 '24 10:01 miquel-espinosa

@coryMosaicML any updates on config ?

Atharva-Phatak avatar Jan 23 '24 00:01 Atharva-Phatak

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.

coryMosaicML avatar Jan 25 '24 21:01 coryMosaicML