aje6 commited on
Commit
7ced0cd
·
verified ·
1 Parent(s): cf7e217

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +39 -2
config.json CHANGED
@@ -1,5 +1,34 @@
1
  {
2
- "_name_or_path": "onnx/Model_IV.onnx",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  "id2label": {
4
  "0": "A",
5
  "1": "B",
@@ -28,6 +57,9 @@
28
  "24": "Y",
29
  "25": "Z"
30
  },
 
 
 
31
  "label2id": {
32
  "A": 0,
33
  "B": 1,
@@ -56,10 +88,15 @@
56
  "Y": 24,
57
  "Z": 25
58
  },
 
59
  "max_position_embeddings": 1024,
60
  "model_type": "detr",
61
  "num_channels": 3,
62
  "num_hidden_layers": 6,
63
  "num_queries": 100,
64
- "use_pretrained_backbone": false,
 
 
 
 
65
  }
 
1
  {
2
+ "_commit_hash": "480370a8aeeed9fc8d78837b4e94e5f936fe73f2",
3
+ "_name_or_path": "facebook/detr-resnet-50",
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "relu",
6
+ "architectures": [
7
+ "DetrForObjectDetection"
8
+ ],
9
+ "attention_dropout": 0.0,
10
+ "auxiliary_loss": false,
11
+ "backbone": "resnet50",
12
+ "backbone_config": null,
13
+ "bbox_cost": 5,
14
+ "bbox_loss_coefficient": 5,
15
+ "class_cost": 1,
16
+ "classifier_dropout": 0.0,
17
+ "d_model": 256,
18
+ "decoder_attention_heads": 8,
19
+ "decoder_ffn_dim": 2048,
20
+ "decoder_layerdrop": 0.0,
21
+ "decoder_layers": 6,
22
+ "dice_loss_coefficient": 1,
23
+ "dilation": false,
24
+ "dropout": 0.1,
25
+ "encoder_attention_heads": 8,
26
+ "encoder_ffn_dim": 2048,
27
+ "encoder_layerdrop": 0.0,
28
+ "encoder_layers": 6,
29
+ "eos_coefficient": 0.1,
30
+ "giou_cost": 2,
31
+ "giou_loss_coefficient": 2,
32
  "id2label": {
33
  "0": "A",
34
  "1": "B",
 
57
  "24": "Y",
58
  "25": "Z"
59
  },
60
+ "init_std": 0.02,
61
+ "init_xavier_std": 1.0,
62
+ "is_encoder_decoder": true,
63
  "label2id": {
64
  "A": 0,
65
  "B": 1,
 
88
  "Y": 24,
89
  "Z": 25
90
  },
91
+ "mask_loss_coefficient": 1,
92
  "max_position_embeddings": 1024,
93
  "model_type": "detr",
94
  "num_channels": 3,
95
  "num_hidden_layers": 6,
96
  "num_queries": 100,
97
+ "position_embedding_type": "sine",
98
+ "scale_embedding": false,
99
+ "transformers_version": "4.7.0.dev0",
100
+ "use_pretrained_backbone": true,
101
+ "use_timm_backbone": true
102
  }