Harley-ml
/

MNIST-IMG-390k

digit_diffusion

Model card Files Files and versions

Harley-ml commited on 4 days ago

Commit

8a8faa7

·

verified ·

1 Parent(s): b0236d2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -37,7 +37,7 @@ MNiST-IMG-390k is an ~**390k parameter model** trained to **generate an image**
 | `in_channels`        | `1`            |
 | `out_channels`       | `1`            |
 | `num_classes`        | `10`           |
-| `block_out_channels` | `[16, 20, 28]` |
 | `layers_per_block`   | `8`            |
 | `norm_num_groups`    | `4`            |

 | `in_channels`        | `1`            |
 | `out_channels`       | `1`            |
 | `num_classes`        | `10`           |
+| `block_out_channels` | `[12, 16, 20]` |
 | `layers_per_block`   | `8`            |
 | `norm_num_groups`    | `4`            |