rafaelpadilla commited on
Commit
a18aecd
1 Parent(s): 6f7feee

Add model from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (2) hide show
  1. config.json +4 -0
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,5 +1,8 @@
1
  {
2
  "act_encoder": "silu",
 
 
 
3
  "aux_loss": true,
4
  "backbone_config": {
5
  "backbone": "resnet50d",
@@ -219,6 +222,7 @@
219
  "num_levels": 3,
220
  "num_queries": 300,
221
  "pe_temperature": 10000,
 
222
  "transformers_version": "4.36.0.dev0",
223
  "use_encoder_idx": [
224
  2
 
1
  {
2
  "act_encoder": "silu",
3
+ "architectures": [
4
+ "RTDetrModel"
5
+ ],
6
  "aux_loss": true,
7
  "backbone_config": {
8
  "backbone": "resnet50d",
 
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
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a7d257dd291aa027058908d8f4410bfb594d2a29989a35ef66de4a5452a259e0
3
- size 172160832
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1bd882b71438fb1bcf40fbf4d8e86c6ba94bdc19481ec0f50a7b70178435e95
3
+ size 172161244