Animadversio / mini_edm

Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist

https://arxiv.org/abs/2206.00364

Geek Repo

Github PK Tool

Mini-EDM -- A minimalistic EDM with all the essence

Introduction

To train or evaluate the EDM, please follow the command in run.sh.

Results

Results FID on CIFAIR-10:

FID of this implementation.

Reference FID (each snapshot=5000k images=5000k/128 steps) from the original EDM codebase using the same model architecture and the same batch size of 128 (nothing else changed):

FID of original EDM.

The difference may come from the learning rate and warmup strategy.

About

Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist

https://arxiv.org/abs/2206.00364

Languages

Language:Python 54.2%Language:Jupyter Notebook 35.1%Language:Shell 10.7%