utsavnandi commited on
Commit
93de90d
1 Parent(s): f11042f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -2,13 +2,13 @@ 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)
 
2
 
3
  Model Hyperparams:
4
 
5
+ - Model size: 51,834,625 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
+ - Total steps: 5,000
13
 
14
  ![output.png](https://s3.amazonaws.com/moonup/production/uploads/1672153152960-6262d89f63f73be3d2f6b7c1.png)