CoffeeGitta commited on
Commit
1d3c400
·
verified ·
1 Parent(s): ff66bbe

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -10
config.json CHANGED
@@ -7,18 +7,17 @@
7
  "attention_dropout": 0.0,
8
  "auxiliary_loss": false,
9
  "backbone": "resnet50",
10
- "backbone_config": null,
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": 1024,
 
17
  "decoder_layers": 6,
18
  "decoder_n_points": 4,
19
  "dice_loss_coefficient": 1,
20
  "dilation": false,
21
- "disable_custom_kernels": false,
22
  "dropout": 0.1,
23
  "encoder_attention_heads": 8,
24
  "encoder_ffn_dim": 1024,
@@ -26,7 +25,6 @@
26
  "encoder_layers": 6,
27
  "encoder_n_points": 4,
28
  "eos_coefficient": 0.1,
29
- "focal_alpha": 0.25,
30
  "giou_cost": 2,
31
  "giou_loss_coefficient": 2,
32
  "id2label": {
@@ -63,15 +61,13 @@
63
  "mask_loss_coefficient": 1,
64
  "max_position_embeddings": 1024,
65
  "model_type": "deformable_detr",
66
- "num_channels": 3,
67
  "num_feature_levels": 4,
68
- "num_queries": 200,
69
  "position_embedding_type": "sine",
 
70
  "torch_dtype": "float32",
71
- "transformers_version": "4.36.2",
72
- "two_stage": true,
73
- "two_stage_num_proposals": 200,
74
- "use_pretrained_backbone": true,
75
- "use_timm_backbone": "SenseTime/deformable-detr-with-box-refine-two-stage",
76
  "with_box_refine": true
77
  }
 
7
  "attention_dropout": 0.0,
8
  "auxiliary_loss": false,
9
  "backbone": "resnet50",
 
10
  "bbox_cost": 5,
11
  "bbox_loss_coefficient": 5,
12
  "class_cost": 1,
13
  "d_model": 256,
14
  "decoder_attention_heads": 8,
15
  "decoder_ffn_dim": 1024,
16
+ "decoder_layerdrop": 0.0,
17
  "decoder_layers": 6,
18
  "decoder_n_points": 4,
19
  "dice_loss_coefficient": 1,
20
  "dilation": false,
 
21
  "dropout": 0.1,
22
  "encoder_attention_heads": 8,
23
  "encoder_ffn_dim": 1024,
 
25
  "encoder_layers": 6,
26
  "encoder_n_points": 4,
27
  "eos_coefficient": 0.1,
 
28
  "giou_cost": 2,
29
  "giou_loss_coefficient": 2,
30
  "id2label": {
 
61
  "mask_loss_coefficient": 1,
62
  "max_position_embeddings": 1024,
63
  "model_type": "deformable_detr",
 
64
  "num_feature_levels": 4,
65
+ "num_queries": 300,
66
  "position_embedding_type": "sine",
67
+ "return_intermediate": true,
68
  "torch_dtype": "float32",
69
+ "transformers_version": "4.16.0.dev0",
70
+ "two_stage": false,
71
+ "two_stage_num_proposals": 300,
 
 
72
  "with_box_refine": true
73
  }