Update README.md
Browse files
README.md
CHANGED
|
@@ -47,9 +47,8 @@ against traditional/manual augmentation techniques for CNN classification.
|
|
| 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,9 +63,8 @@ against traditional/manual augmentation techniques for CNN classification.
|
|
| 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)
|
|
|
|
| 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 |
- EMA: Disabled
|
| 64 |
- Diffusion: 1000 Timesteps (Linear Schedule)
|
| 65 |
|
| 66 |
+
==========
|
| 67 |
+
LATENT DDPM (LDM) ===========================================================
|
|
|
|
| 68 |
• Optimization Strategy:
|
| 69 |
- Optimizer: AdamW
|
| 70 |
- Learning Rate: 1e-4 (with Cosine Annealing)
|