search

Warvito / generative_chestxray

Repository to train Latent Diffusion Models on Chest X-ray data (MIMIC-CXR) using MONAI Generative Models
Apache License 2.0
51 stars 10 forks source link

Improve the AutoencoderKL model #1

Open Warvito opened 1 year ago

Warvito commented 1 year ago

Improve the Autoencoder KL model using the Stable Diffusion approach for the adversarial component and using a better learning rate scheduling.

Warvito commented 1 year ago

In order to improve the Autoencoder KL performance, I decided to increase the number of parameters of it for now