resberry commited on
Commit
45045d2
·
1 Parent(s): 82767a2

Initial model upload

Browse files
Files changed (3) hide show
  1. README.md +3 -0
  2. config.json +10 -0
  3. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ # MNIST CNN Classifier
2
+ This is a CNN model trained on the MNIST dataset for digit classification.
3
+
config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "input_shape": [
3
+ 1,
4
+ 28,
5
+ 28
6
+ ],
7
+ "num_classes": 10,
8
+ "framework": "pytorch",
9
+ "model_type": "cnn"
10
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9f7f6d993a2eb49dfacc8169c935e3138a3dfd50229d9584d8324d5cb1df496
3
+ size 6506824