ronchern commited on
Commit
fee1a34
·
verified ·
1 Parent(s): a1eef89

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -8
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
- • Optimization Strategy:
 
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