utsavnandi commited on
Commit
f11042f
1 Parent(s): 83d92dc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -6
README.md CHANGED
@@ -1,11 +1,14 @@
1
  Fashion MNIST unconditional Unet Model trained using DDPM
2
 
3
- Model size: 51834625 params
4
-
5
- Optimizer: Adam
6
- LR: 3e-4
7
- Batch Size: 64
8
- Grad Accumulation: 8 steps
 
 
 
9
 
10
 
11
  ![output.png](https://s3.amazonaws.com/moonup/production/uploads/1672153152960-6262d89f63f73be3d2f6b7c1.png)
 
1
  Fashion MNIST unconditional Unet Model trained using DDPM
2
 
3
+ Model Hyperparams:
4
+
5
+ - Model size: 51834625 params
6
+ - 3 stages: 128, 256, 512 channels
7
+ - Linear Attention in 2nd and 3rd stages, Self Attention in Middle Stage
8
+ - Optimizer: Adam
9
+ - LR: 3e-4
10
+ - Batch Size: 64
11
+ - Grad Accumulation: 8 steps
12
 
13
 
14
  ![output.png](https://s3.amazonaws.com/moonup/production/uploads/1672153152960-6262d89f63f73be3d2f6b7c1.png)