Harley-ml commited on
Commit
13351e2
·
verified ·
1 Parent(s): 20f74c7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -26,32 +26,32 @@ Framework: PyTorch, diffusers
26
  Author: Paul Courneya (Harley-ml)
27
  ```
28
 
29
- ## Description
30
  MNiST-IMG-390k is an ~**390k parameter model** trained to **generate an image** based on an **input number (0-9)**.
31
 
32
- ## Training
33
 
34
- ### Hardware
35
 
36
  MNiST-IMG was trained on Google Colaboratory (NVIDA Tesla T4) for ~10 minutes with a batch size of 64 for 10 epochs.
37
 
38
- ### Dataset
39
 
40
  [ylecun/mnist](https://huggingface.co/ylecun/mnist)
41
 
42
- ### Training Results
43
 
44
- Loss ended at ~0.39.
45
 
46
  Note: I can't provide the raw training logs as I loss it somehwere after training. Sorry!
47
 
48
  ## **Generation Examples**
49
 
50
- At 1000 decoding steps:
51
 
52
  ![1000 Decoding Step Digit Image Generation](images/digit_image_samples_1000s.png)
53
 
54
- At 200 decoding steps:
55
 
56
  ![1000 Decoding Step Generation Image](images/digit_image_samples_200s.png)
57
 
 
26
  Author: Paul Courneya (Harley-ml)
27
  ```
28
 
29
+ ## **Description**
30
  MNiST-IMG-390k is an ~**390k parameter model** trained to **generate an image** based on an **input number (0-9)**.
31
 
32
+ ## **Training**
33
 
34
+ ### **Hardware**
35
 
36
  MNiST-IMG was trained on Google Colaboratory (NVIDA Tesla T4) for ~10 minutes with a batch size of 64 for 10 epochs.
37
 
38
+ ### **Dataset**
39
 
40
  [ylecun/mnist](https://huggingface.co/ylecun/mnist)
41
 
42
+ ### **Training Results**
43
 
44
+ Loss ended at ~**0.39**.
45
 
46
  Note: I can't provide the raw training logs as I loss it somehwere after training. Sorry!
47
 
48
  ## **Generation Examples**
49
 
50
+ At **1000** decoding steps:
51
 
52
  ![1000 Decoding Step Digit Image Generation](images/digit_image_samples_1000s.png)
53
 
54
+ At **200** decoding steps:
55
 
56
  ![1000 Decoding Step Generation Image](images/digit_image_samples_200s.png)
57