Document Diffusion artifacts
Browse files
README.md
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: pytorch
|
| 3 |
+
tags:
|
| 4 |
+
- diffusion
|
| 5 |
+
- ddpm
|
| 6 |
+
- cifar10
|
| 7 |
+
---
|
| 8 |
+
# Diffusion training outputs
|
| 9 |
+
|
| 10 |
+
Unconditional and conditional DDPM training artifacts for CIFAR-10, using
|
| 11 |
+
custom UNet architectures with attention and EMA.
|
| 12 |
+
|
| 13 |
+
Each experiment directory contains PyTorch checkpoints, generated sample images,
|
| 14 |
+
and a training history CSV. Checkpoints contain model, EMA model, optimizer,
|
| 15 |
+
scheduler, epoch, and loss state. Use the matching code from the Diffusion
|
| 16 |
+
project to load them; these are custom PyTorch models.
|
| 17 |
+
|
| 18 |
+
Download the repository contents into `Unet/outputs` in the code project.
|