iammbrn commited on
Commit
ddc93c9
·
verified ·
1 Parent(s): ee60403

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +20 -0
config.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": ["ResNet50"],
3
+ "model_type": "resnet",
4
+ "num_labels": 5,
5
+ "label2id": {
6
+ "Alkol": 0,
7
+ "Normal": 1,
8
+ "NSFW": 2,
9
+ "Silah": 3,
10
+ "Tutun": 4
11
+ },
12
+ "id2label": {
13
+ "0": "Alkol",
14
+ "1": "Normal",
15
+ "2": "NSFW",
16
+ "3": "Silah",
17
+ "4": "Tutun"
18
+ },
19
+ "input_shape": [224, 224, 3]
20
+ }