Training in progress, epoch 1

config.json

@@ -1,5 +1,4 @@
 {
-  "_name_or_path": "vladislavbro/dfine_l_coco",
   "activation_dropout": 0.0,
   "activation_function": "silu",
   "anchor_image_size": null,
@@ -10,9 +9,25 @@
   "auxiliary_loss": true,
   "backbone": null,
   "backbone_config": {
+    "depths": [
+      3,
+      4,
+      6,
+      3
+    ],
+    "downsample_in_bottleneck": false,
+    "downsample_in_first_stage": false,
     "embedding_size": 32,
+    "hidden_act": "relu",
+    "hidden_sizes": [
+      256,
+      512,
+      1024,
+      2048
+    ],
     "layer_type": "basic",
-    "model_type": "d_fine_resnet",
+    "model_type": "hgnet_v2",
+    "num_channels": 3,
     "out_features": [
       "stage2",
       "stage3",
@@ -22,7 +37,68 @@
       2,
       3,
       4
-    ]
+    ],
+    "stage_downsample": [
+      false,
+      true,
+      true,
+      true
+    ],
+    "stage_in_channels": [
+      48,
+      128,
+      512,
+      1024
+    ],
+    "stage_kernel_size": [
+      3,
+      3,
+      5,
+      5
+    ],
+    "stage_light_block": [
+      false,
+      false,
+      true,
+      true
+    ],
+    "stage_mid_channels": [
+      48,
+      96,
+      192,
+      384
+    ],
+    "stage_names": [
+      "stem",
+      "stage1",
+      "stage2",
+      "stage3",
+      "stage4"
+    ],
+    "stage_num_blocks": [
+      1,
+      1,
+      3,
+      1
+    ],
+    "stage_numb_of_layers": [
+      6,
+      6,
+      6,
+      6
+    ],
+    "stage_out_channels": [
+      128,
+      512,
+      1024,
+      2048
+    ],
+    "stem_channels": [
+      3,
+      32,
+      48
+    ],
+    "use_learnable_affine_block": false
   },
   "backbone_kwargs": null,
   "batch_norm_eps": 1e-05,
@@ -45,7 +121,6 @@
   ],
   "decoder_offset_scale": 0.5,
   "depth_mult": 1.0,
-  "disable_custom_kernels": true,
   "dropout": 0.0,
   "encode_proj_layers": [
     2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:59b50bf70dcd9f7aeeaac199b1113176eec82d78354ac0d5c133faa05f8feefe
-size 124999228
+oid sha256:879d9ceaca1f357db061b7b9ef56b5fda6c89ad8dfd25380977b14233918e501
+size 125001980

preprocessor_config.json

@@ -1,6 +1,8 @@
 {
   "crop_size": null,
+  "data_format": "channels_first",
   "default_to_square": false,
+  "device": null,
   "do_center_crop": null,
   "do_convert_annotations": true,
   "do_convert_rgb": null,
@@ -20,9 +22,12 @@
     0.224,
     0.225
   ],
+  "input_data_format": null,
   "pad_size": null,
   "resample": 2,
   "rescale_factor": 0.00392156862745098,
+  "return_segmentation_masks": null,
+  "return_tensors": null,
   "size": {
     "height": 480,
     "width": 480

runs/Mar18_16-21-26_15a84bf44b1b/events.out.tfevents.1742314888.15a84bf44b1b.8417.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18394c9df76a62d309f6491abf008eff8ff9b99d4b435026b71ccdec1b7ca6e0
+size 9508

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8d6cb829cddfd4502d29f1c0c1b54f94954a8442d0dd393131b198629eece2b4
+oid sha256:19382aa91092f96579f434f8e11964c1fb03a39463d05fd754536efb99e65a1e
 size 5432