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
49 stars 10 forks source link

Improve Diffusion Model #2

Open Warvito opened 1 year ago

Warvito commented 1 year ago

Improve Diffusion Model by using a better learning rate scheduler (including warm-up period) and using exponential moving average.