config.json
1.4 KB · 60 lines · json Raw
1 {
2 "activation_dropout": 0.0,
3 "activation_function": "relu",
4 "architectures": [
5 "TableTransformerForObjectDetection"
6 ],
7 "attention_dropout": 0.0,
8 "auxiliary_loss": false,
9 "backbone": "resnet18",
10 "bbox_cost": 5,
11 "bbox_loss_coefficient": 5,
12 "ce_loss_coefficient": 1,
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.4,
27 "giou_cost": 2,
28 "giou_loss_coefficient": 2,
29 "id2label": {
30 "0": "table",
31 "1": "table column",
32 "2": "table row",
33 "3": "table column header",
34 "4": "table projected row header",
35 "5": "table spanning cell"
36 },
37 "init_std": 0.02,
38 "init_xavier_std": 1.0,
39 "is_encoder_decoder": true,
40 "label2id": {
41 "table": 0,
42 "table column": 1,
43 "table column header": 3,
44 "table projected row header": 4,
45 "table row": 2,
46 "table spanning cell": 5
47 },
48 "mask_loss_coefficient": 1,
49 "max_position_embeddings": 1024,
50 "model_type": "table-transformer",
51 "num_channels": 3,
52 "num_hidden_layers": 6,
53 "num_queries": 125,
54 "position_embedding_type": "sine",
55 "scale_embedding": false,
56 "torch_dtype": "float32",
57 "transformers_version": "4.24.0.dev0",
58 "use_pretrained_backbone": true
59 }
60