Update README.md
Browse files
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 |

|
| 53 |
|
| 54 |
-
At 200 decoding steps:
|
| 55 |
|
| 56 |

|
| 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 |

|
| 53 |
|
| 54 |
+
At **200** decoding steps:
|
| 55 |
|
| 56 |

|
| 57 |
|