Update README.md
Browse files
README.md
CHANGED
|
@@ -47,8 +47,9 @@ against traditional/manual augmentation techniques for CNN classification.
|
|
| 47 |
|
| 48 |
#### Training Hyperparameters
|
| 49 |
|
| 50 |
-
==========
|
| 51 |
-
PIXEL DDPM (Baseline Model) ==================
|
|
|
|
| 52 |
• Optimization Strategy:
|
| 53 |
- Optimizer: AdamW
|
| 54 |
- Learning Rate: 2e-4 (with Cosine Annealing)
|
|
@@ -63,9 +64,10 @@ against traditional/manual augmentation techniques for CNN classification.
|
|
| 63 |
- EMA: Disabled
|
| 64 |
- Diffusion: 1000 Timesteps (Linear Schedule)
|
| 65 |
|
| 66 |
-
==========
|
| 67 |
-
LATENT DDPM (LDM) ============================
|
| 68 |
-
|
|
|
|
| 69 |
- Optimizer: AdamW
|
| 70 |
- Learning Rate: 1e-4 (with Cosine Annealing)
|
| 71 |
- Weight Decay: 1e-3
|
|
@@ -78,9 +80,6 @@ against traditional/manual augmentation techniques for CNN classification.
|
|
| 78 |
- Channel Mults: (2, 2)
|
| 79 |
- EMA: Enabled (Decay: 0.99)
|
| 80 |
|
| 81 |
-
• Conditioning (CFG):
|
| 82 |
-
- Label Dropout: 15% (during training)
|
| 83 |
-
- Guidance Scale: 5.0 (during inference)
|
| 84 |
===========================================================
|
| 85 |
|
| 86 |
|
|
|
|
| 47 |
|
| 48 |
#### Training Hyperparameters
|
| 49 |
|
| 50 |
+
===================
|
| 51 |
+
PIXEL DDPM (Baseline Model) ==================
|
| 52 |
+
|
| 53 |
• Optimization Strategy:
|
| 54 |
- Optimizer: AdamW
|
| 55 |
- Learning Rate: 2e-4 (with Cosine Annealing)
|
|
|
|
| 64 |
- EMA: Disabled
|
| 65 |
- Diffusion: 1000 Timesteps (Linear Schedule)
|
| 66 |
|
| 67 |
+
===================
|
| 68 |
+
LATENT DDPM (LDM) ============================
|
| 69 |
+
|
| 70 |
+
• Optimization Strategy:
|
| 71 |
- Optimizer: AdamW
|
| 72 |
- Learning Rate: 1e-4 (with Cosine Annealing)
|
| 73 |
- Weight Decay: 1e-3
|
|
|
|
| 80 |
- Channel Mults: (2, 2)
|
| 81 |
- EMA: Enabled (Decay: 0.99)
|
| 82 |
|
|
|
|
|
|
|
|
|
|
| 83 |
===========================================================
|
| 84 |
|
| 85 |
|