Sebi1128 / DAL_CS

1 stars 0 forks source link

Dev/vae training #11

Closed AronDiSc closed 2 years ago

AronDiSc commented 2 years ago

Added vae pre training and the possibility to load such pre-trained off the shelf vaes

What now doesn't work anymore is using the 256 CIFAR10 vae from lightning bolts. If it is needed we could quickly load the model in the python shell and safe the model parameters as .ckpt files