Training in progress, epoch 1

config.json

@@ -11,16 +11,21 @@
   ],
   "attention_dropout": 0.0,
   "auxiliary_loss": true,
-  "backbone": "resnet50d",
-  "backbone_config": null,
-  "backbone_kwargs": {
-    "features_only": true,
+  "backbone": null,
+  "backbone_config": {
+    "model_type": "rt_detr_resnet",
+    "out_features": [
+      "stage2",
+      "stage3",
+      "stage4"
+    ],
     "out_indices": [
       2,
       3,
       4
     ]
   },
+  "backbone_kwargs": null,
   "batch_norm_eps": 1e-05,
   "box_noise_scale": 1.0,
   "d_model": 256,
@@ -92,8 +97,8 @@
   "torch_dtype": "float32",
   "transformers_version": "4.42.0.dev0",
   "use_focal_loss": true,
-  "use_pretrained_backbone": true,
-  "use_timm_backbone": true,
+  "use_pretrained_backbone": false,
+  "use_timm_backbone": false,
   "weight_loss_bbox": 5.0,
   "weight_loss_giou": 2.0,
   "weight_loss_vfl": 1.0,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e594acadd010fa6ca304a28b086587f25f9a54c8eaee5240b1bf191e8eede739
-size 171552884
+oid sha256:4b7a58a102c59881ecf8ecb80638d75bd51e9b5ed7c69a7066c6c54348361ea3
+size 171559340

preprocessor_config.json

@@ -14,6 +14,7 @@
     "image_mean",
     "image_std",
     "do_pad",
+    "pad_size",
     "format",
     "return_tensors",
     "data_format",
@@ -36,6 +37,7 @@
     0.224,
     0.225
   ],
+  "pad_size": null,
   "resample": 2,
   "rescale_factor": 0.00392156862745098,
   "size": {

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:05d2d9d1722da01089ac23e156be106e8c6defb22a144f3b0cf71f7a90408ec0
+oid sha256:b53e1367e265a694e532fa4ff1611f43c488c6de800fe4ac2d41c059fe9bf91c
 size 5176