Update README.md
Browse files
README.md
CHANGED
|
@@ -30,6 +30,19 @@ Author: SupraLabs
|
|
| 30 |
## **Description**
|
| 31 |
MNiST-IMG-200k is an ~**200k parameter model** trained to **generate an image** based on an **input number (0-9)**.
|
| 32 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
## **Training**
|
| 34 |
|
| 35 |
### **Hardware**
|
|
|
|
| 30 |
## **Description**
|
| 31 |
MNiST-IMG-200k is an ~**200k parameter model** trained to **generate an image** based on an **input number (0-9)**.
|
| 32 |
|
| 33 |
+
## Architecture
|
| 34 |
+
|
| 35 |
+
| Parameter | Value |
|
| 36 |
+
| -------------------- | ---------- |
|
| 37 |
+
| `image_size` | `32` |
|
| 38 |
+
| `sample_size` | `32` |
|
| 39 |
+
| `in_channels` | `1` |
|
| 40 |
+
| `out_channels` | `1` |
|
| 41 |
+
| `num_classes` | `10` |
|
| 42 |
+
| `block_out_channels` | `[12, 16]` |
|
| 43 |
+
| `layers_per_block` | `8` |
|
| 44 |
+
| `norm_num_groups` | `4` |
|
| 45 |
+
|
| 46 |
## **Training**
|
| 47 |
|
| 48 |
### **Hardware**
|