generative_chestxray
generative_chestxray copied to clipboard
Improve the AutoencoderKL model
Improve the Autoencoder KL model using the Stable Diffusion approach for the adversarial component and using a better learning rate scheduling.
In order to improve the Autoencoder KL performance, I decided to increase the number of parameters of it for now