PITICHA commited on
Commit
e8ae932
·
verified ·
1 Parent(s): 924e86f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +14 -10
config.json CHANGED
@@ -1,10 +1,14 @@
1
- {
2
- "architectures": [
3
- "MaskRCNNForInstanceSegmentation"
4
- ],
5
- "dtype": "float32",
6
- "hidden_layer": 256,
7
- "model_type": "maskrcnn",
8
- "num_classes": 11,
9
- "transformers_version": "4.57.3"
10
- }
 
 
 
 
 
1
+ {
2
+ "architectures": ["MaskRCNNForInstanceSegmentation"],
3
+ "model_type": "maskrcnn",
4
+ "num_classes": 11,
5
+ "hidden_layer": 256,
6
+ "dtype": "float32",
7
+
8
+ "auto_map": {
9
+ "AutoConfig": "model.configuration_maskrcnn.MaskRCNNConfig",
10
+ "AutoModel": "model.modeling_maskrcnn.MaskRCNNForInstanceSegmentation"
11
+ },
12
+
13
+ "transformers_version": "4.57.3"
14
+ }