Bono93 commited on
Commit
e859782
·
1 Parent(s): 68cd5f5

Upload SimpleCNNForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +50 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "SimpleCNNForImageClassification"
4
+ ],
5
+ "conv1_shape": [
6
+ 1,
7
+ 32,
8
+ 3,
9
+ 1
10
+ ],
11
+ "conv2_shape": [
12
+ 32,
13
+ 64,
14
+ 3,
15
+ 1
16
+ ],
17
+ "dropout1": 0.25,
18
+ "dropout2": 0.5,
19
+ "fc_shape": [
20
+ 9216,
21
+ 128
22
+ ],
23
+ "id2label": {
24
+ "0": "LABEL_0",
25
+ "1": "LABEL_1",
26
+ "2": "LABEL_2",
27
+ "3": "LABEL_3",
28
+ "4": "LABEL_4",
29
+ "5": "LABEL_5",
30
+ "6": "LABEL_6",
31
+ "7": "LABEL_7",
32
+ "8": "LABEL_8",
33
+ "9": "LABEL_9"
34
+ },
35
+ "label2id": {
36
+ "LABEL_0": 0,
37
+ "LABEL_1": 1,
38
+ "LABEL_2": 2,
39
+ "LABEL_3": 3,
40
+ "LABEL_4": 4,
41
+ "LABEL_5": 5,
42
+ "LABEL_6": 6,
43
+ "LABEL_7": 7,
44
+ "LABEL_8": 8,
45
+ "LABEL_9": 9
46
+ },
47
+ "model_type": "simplecnn",
48
+ "torch_dtype": "float32",
49
+ "transformers_version": "4.27.4"
50
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a299be80d4165a4a67602dee6b67a3998351dc49829fbb73eb000098185c115
3
+ size 9598171