config.json
1.1 KB · 50 lines · json Raw
1 {
2 "_name_or_path": "facebook/detr-resnet-50",
3 "activation_dropout": 0.0,
4 "activation_function": "relu",
5 "architectures": [
6 "DetrForObjectDetection"
7 ],
8 "attention_dropout": 0.0,
9 "auxiliary_loss": false,
10 "backbone": "resnet50",
11 "bbox_cost": 5,
12 "bbox_loss_coefficient": 5,
13 "class_cost": 1,
14 "d_model": 256,
15 "decoder_attention_heads": 8,
16 "decoder_ffn_dim": 2048,
17 "decoder_layerdrop": 0.0,
18 "decoder_layers": 6,
19 "dice_loss_coefficient": 1,
20 "dilation": false,
21 "dropout": 0.1,
22 "encoder_attention_heads": 8,
23 "encoder_ffn_dim": 2048,
24 "encoder_layerdrop": 0.0,
25 "encoder_layers": 6,
26 "eos_coefficient": 0.1,
27 "giou_cost": 2,
28 "giou_loss_coefficient": 2,
29 "id2label": {
30 "0": "table",
31 "1": "table"
32 },
33 "init_std": 0.02,
34 "init_xavier_std": 1.0,
35 "is_encoder_decoder": true,
36 "label2id": {
37 "table": 1,
38 "table": 0
39 },
40 "mask_loss_coefficient": 1,
41 "max_position_embeddings": 1024,
42 "model_type": "detr",
43 "num_hidden_layers": 6,
44 "num_queries": 100,
45 "position_embedding_type": "sine",
46 "scale_embedding": false,
47 "torch_dtype": "float32",
48 "transformers_version": "4.17.0"
49 }
50