rafaelpadilla
commited on
Commit
·
68dd376
1
Parent(s):
2d00ae0
Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py
Browse files- config.json +5 -5
config.json
CHANGED
@@ -1,9 +1,5 @@
|
|
1 |
{
|
2 |
"act_encoder": "silu",
|
3 |
-
"architectures": [
|
4 |
-
"RTDetrModel"
|
5 |
-
],
|
6 |
-
"aux_loss": true,
|
7 |
"backbone_config": {
|
8 |
"backbone": "resnet50d",
|
9 |
"freeze_batch_norm_2d": true,
|
@@ -121,6 +117,10 @@
|
|
121 |
"8": "boat",
|
122 |
"9": "traffic light"
|
123 |
},
|
|
|
|
|
|
|
|
|
124 |
"initializer_range": 0.02,
|
125 |
"label2id": {
|
126 |
"airplane": 4,
|
@@ -222,8 +222,8 @@
|
|
222 |
"num_levels": 3,
|
223 |
"num_queries": 300,
|
224 |
"pe_temperature": 10000,
|
225 |
-
"torch_dtype": "float32",
|
226 |
"transformers_version": "4.36.0.dev0",
|
|
|
227 |
"use_encoder_idx": [
|
228 |
2
|
229 |
],
|
|
|
1 |
{
|
2 |
"act_encoder": "silu",
|
|
|
|
|
|
|
|
|
3 |
"backbone_config": {
|
4 |
"backbone": "resnet50d",
|
5 |
"freeze_batch_norm_2d": true,
|
|
|
117 |
"8": "boat",
|
118 |
"9": "traffic light"
|
119 |
},
|
120 |
+
"image_size": [
|
121 |
+
640,
|
122 |
+
640
|
123 |
+
],
|
124 |
"initializer_range": 0.02,
|
125 |
"label2id": {
|
126 |
"airplane": 4,
|
|
|
222 |
"num_levels": 3,
|
223 |
"num_queries": 300,
|
224 |
"pe_temperature": 10000,
|
|
|
225 |
"transformers_version": "4.36.0.dev0",
|
226 |
+
"use_aux_loss": true,
|
227 |
"use_encoder_idx": [
|
228 |
2
|
229 |
],
|