File size: 923 Bytes
dfb3d01 19e739b dfb3d01 19e739b dfb3d01 299e57c 19e739b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | ---
library_name: pytorch
tags:
- diffusion
- ddpm
- cifar10
---
# Diffusion final checkpoints
Unconditional and conditional DDPM training artifacts for CIFAR-10, using
custom UNet architectures with attention and EMA.
Only the final unconditional and conditional checkpoints are published:
- `ddpm_with_attention_ema/checkpoints/ddpm_final.pt`
- `conditional_ddpm_with_attention_ema/checkpoints/conditional_ddpm_final.pt`
Each file is about 3.11 GiB, about 6.2 GiB combined. Intermediate checkpoints,
sample images, and training history CSV files are not published. The original
full final checkpoints contain model, EMA model, optimizer, scheduler, epoch,
and loss state. Use `ema_model_state_dict` for EMA inference weights, with the
matching custom architectures from the code repository.
Download the repository contents into `Unet/outputs` in the code project.
## Code
https://github.com/wangwenbinw/Diffusion
|