To train or evaluate the EDM, please follow the command in run.sh.
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.