sbchoi commited on
Commit
a060b72
1 Parent(s): dbd8000

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -3
config.json CHANGED
@@ -6,8 +6,9 @@
6
  "DetaForObjectDetection"
7
  ],
8
  "assign_first_stage": true,
 
9
  "attention_dropout": 0.0,
10
- "auxiliary_loss": false,
11
  "backbone_config": {
12
  "_name_or_path": "",
13
  "add_cross_attention": false,
@@ -322,6 +323,6 @@
322
  "torch_dtype": "float32",
323
  "transformers_version": null,
324
  "two_stage": true,
325
- "two_stage_num_proposals": 300,
326
  "with_box_refine": true
327
- }
 
6
  "DetaForObjectDetection"
7
  ],
8
  "assign_first_stage": true,
9
+ "assign_second_stage": true,
10
  "attention_dropout": 0.0,
11
+ "auxiliary_loss": true,
12
  "backbone_config": {
13
  "_name_or_path": "",
14
  "add_cross_attention": false,
 
323
  "torch_dtype": "float32",
324
  "transformers_version": null,
325
  "two_stage": true,
326
+ "two_stage_num_proposals": 900,
327
  "with_box_refine": true
328
+ }