rafaelpadilla commited on
Commit
89a5564
·
1 Parent(s): 92da58d

Add model from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -1,6 +1,9 @@
1
  {
2
  "act_decoder": "relu",
3
  "act_encoder": "silu",
 
 
 
4
  "aux_loss": true,
5
  "backbone_config": {
6
  "backbone": "resnet50d",
@@ -229,6 +232,7 @@
229
  "num_queries": 300,
230
  "pe_temperature": 10000,
231
  "position_embed_type": "sine",
 
232
  "transformers_version": "4.36.0.dev0",
233
  "use_encoder_idx": [
234
  2
 
1
  {
2
  "act_decoder": "relu",
3
  "act_encoder": "silu",
4
+ "architectures": [
5
+ "RTDetrModel"
6
+ ],
7
  "aux_loss": true,
8
  "backbone_config": {
9
  "backbone": "resnet50d",
 
232
  "num_queries": 300,
233
  "pe_temperature": 10000,
234
  "position_embed_type": "sine",
235
+ "torch_dtype": "float32",
236
  "transformers_version": "4.36.0.dev0",
237
  "use_encoder_idx": [
238
  2