dan-lara commited on
Commit
d480f35
·
verified ·
1 Parent(s): 795ffb7

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +35 -0
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ResNetForImageClassification"
4
+ ],
5
+ "depths": [
6
+ 3,
7
+ 4,
8
+ 6,
9
+ 3
10
+ ],
11
+ "downsample_in_first_stage": false,
12
+ "embedding_size": 64,
13
+ "hidden_act": "relu",
14
+ "hidden_sizes": [
15
+ 256,
16
+ 512,
17
+ 1024,
18
+ 2048
19
+ ],
20
+ "id2label": {
21
+ "0": "Batterie",
22
+ "1": "Carton",
23
+ "2": "Metal",
24
+ "3": "Organique",
25
+ "4": "Papier",
26
+ "5": "Plastique",
27
+ "6": "Verre",
28
+ "7": "Vetements"
29
+ },
30
+ "layer_type": "bottleneck",
31
+ "model_type": "resnet",
32
+ "num_channels": 3,
33
+ "torch_dtype": "float32",
34
+ "transformers_version": "4.47.0"
35
+ }