Upload Yolov6ForObjectDetection
Browse files- config.json +2 -1
config.json
CHANGED
@@ -228,7 +228,8 @@
|
|
228 |
"neck_stage_block_type": "Yolov6RepBlock",
|
229 |
"num_attention_heads": 3,
|
230 |
"num_hidden_layers": 12,
|
231 |
-
"reg_max":
|
|
|
232 |
"torch_dtype": "float32",
|
233 |
"transformers_version": "4.37.0.dev0",
|
234 |
"use_dfl": false,
|
|
|
228 |
"neck_stage_block_type": "Yolov6RepBlock",
|
229 |
"num_attention_heads": 3,
|
230 |
"num_hidden_layers": 12,
|
231 |
+
"reg_max": 0,
|
232 |
+
"reg_max_proj": 16,
|
233 |
"torch_dtype": "float32",
|
234 |
"transformers_version": "4.37.0.dev0",
|
235 |
"use_dfl": false,
|