Upload 4 files

deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k.json

@@ -0,0 +1,70 @@
+{
+  "birder_version": "v0.0.8a11",
+  "name": "deformable_detr",
+  "alias": "deformable_detr_boxref",
+  "task": "object_detection",
+  "net_param": null,
+  "model_config": {
+    "box_refine": true
+  },
+  "backbone": "convnext_v2",
+  "backbone_alias": "convnext_v2_tiny",
+  "backbone_config": {
+    "in_channels": [
+      96,
+      192,
+      384,
+      768
+    ],
+    "num_layers": [
+      3,
+      3,
+      9,
+      3
+    ],
+    "drop_path_rate": 0.2
+  },
+  "signature": {
+    "dynamic": true,
+    "inputs": [
+      {
+        "data_shape": [
+          0,
+          3,
+          640,
+          640
+        ]
+      }
+    ],
+    "outputs": [
+      [
+        {
+          "boxes": [
+            0,
+            4
+          ],
+          "labels": [
+            0
+          ],
+          "scores": [
+            0
+          ]
+        }
+      ],
+      {}
+    ],
+    "num_labels": 91
+  },
+  "rgb_stats": {
+    "mean": [
+      0.5191,
+      0.5306,
+      0.4877
+    ],
+    "std": [
+      0.2316,
+      0.2304,
+      0.2588
+    ]
+  }
+}

deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k.pt

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

training.log

@@ -0,0 +1,603 @@
+Starting training with learning rate of 0.0002
+Epoch 1/50 training_loss: 14.2889
+Epoch 1/50 map: 0.1288
+Epoch 1/50 map_small: 0.0236
+Epoch 1/50 map_medium: 0.0883
+Epoch 1/50 map_large: 0.2031
+Epoch 1/50 map_50: 0.2472
+Epoch 1/50 map_75: 0.1175
+Epoch 1/50 mar_1: 0.1910
+Epoch 1/50 mar_10: 0.3288
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_1.pt...
+Time cost: 207m14.1s
+---
+Epoch 2/50 training_loss: 10.6919
+Epoch 2/50 map: 0.2199
+Epoch 2/50 map_small: 0.0367
+Epoch 2/50 map_medium: 0.1461
+Epoch 2/50 map_large: 0.3227
+Epoch 2/50 map_50: 0.3756
+Epoch 2/50 map_75: 0.2221
+Epoch 2/50 mar_1: 0.2401
+Epoch 2/50 mar_10: 0.4034
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_2.pt...
+Time cost: 205m43.3s
+---
+Epoch 3/50 training_loss: 9.6609
+Epoch 3/50 map: 0.2580
+Epoch 3/50 map_small: 0.0469
+Epoch 3/50 map_medium: 0.1671
+Epoch 3/50 map_large: 0.3816
+Epoch 3/50 map_50: 0.4301
+Epoch 3/50 map_75: 0.2654
+Epoch 3/50 mar_1: 0.2597
+Epoch 3/50 mar_10: 0.4287
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_3.pt...
+Time cost: 216m09.6s
+---
+Epoch 4/50 training_loss: 9.1536
+Epoch 4/50 map: 0.2917
+Epoch 4/50 map_small: 0.0544
+Epoch 4/50 map_medium: 0.1910
+Epoch 4/50 map_large: 0.4243
+Epoch 4/50 map_50: 0.4687
+Epoch 4/50 map_75: 0.3045
+Epoch 4/50 mar_1: 0.2761
+Epoch 4/50 mar_10: 0.4562
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_4.pt...
+Time cost: 216m12.6s
+---
+Epoch 5/50 training_loss: 8.7942
+Epoch 5/50 map: 0.3062
+Epoch 5/50 map_small: 0.0630
+Epoch 5/50 map_medium: 0.2000
+Epoch 5/50 map_large: 0.4456
+Epoch 5/50 map_50: 0.4817
+Epoch 5/50 map_75: 0.3240
+Epoch 5/50 mar_1: 0.2871
+Epoch 5/50 mar_10: 0.4667
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_5.pt...
+Time cost: 216m49.2s
+---
+Epoch 6/50 training_loss: 8.5256
+Epoch 6/50 map: 0.3224
+Epoch 6/50 map_small: 0.0663
+Epoch 6/50 map_medium: 0.2158
+Epoch 6/50 map_large: 0.4632
+Epoch 6/50 map_50: 0.5006
+Epoch 6/50 map_75: 0.3396
+Epoch 6/50 mar_1: 0.2975
+Epoch 6/50 mar_10: 0.4823
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_6.pt...
+Time cost: 216m40.7s
+---
+Epoch 7/50 training_loss: 8.3521
+Epoch 7/50 map: 0.3432
+Epoch 7/50 map_small: 0.0732
+Epoch 7/50 map_medium: 0.2287
+Epoch 7/50 map_large: 0.4910
+Epoch 7/50 map_50: 0.5261
+Epoch 7/50 map_75: 0.3645
+Epoch 7/50 mar_1: 0.3071
+Epoch 7/50 mar_10: 0.4929
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_7.pt...
+Time cost: 216m51.8s
+---
+Epoch 8/50 training_loss: 8.1895
+Epoch 8/50 map: 0.3462
+Epoch 8/50 map_small: 0.0751
+Epoch 8/50 map_medium: 0.2362
+Epoch 8/50 map_large: 0.4942
+Epoch 8/50 map_50: 0.5313
+Epoch 8/50 map_75: 0.3670
+Epoch 8/50 mar_1: 0.3095
+Epoch 8/50 mar_10: 0.4970
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_8.pt...
+Time cost: 216m53.3s
+---
+Epoch 9/50 training_loss: 8.0642
+Epoch 9/50 map: 0.3456
+Epoch 9/50 map_small: 0.0698
+Epoch 9/50 map_medium: 0.2341
+Epoch 9/50 map_large: 0.4962
+Epoch 9/50 map_50: 0.5353
+Epoch 9/50 map_75: 0.3667
+Epoch 9/50 mar_1: 0.3077
+Epoch 9/50 mar_10: 0.4899
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_9.pt...
+Time cost: 217m44.1s
+---
+Epoch 10/50 training_loss: 7.9581
+Epoch 10/50 map: 0.3585
+Epoch 10/50 map_small: 0.0724
+Epoch 10/50 map_medium: 0.2449
+Epoch 10/50 map_large: 0.5113
+Epoch 10/50 map_50: 0.5468
+Epoch 10/50 map_75: 0.3838
+Epoch 10/50 mar_1: 0.3117
+Epoch 10/50 mar_10: 0.5011
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_10.pt...
+Time cost: 216m46.2s
+---
+Epoch 11/50 training_loss: 7.8656
+Epoch 11/50 map: 0.3631
+Epoch 11/50 map_small: 0.0806
+Epoch 11/50 map_medium: 0.2498
+Epoch 11/50 map_large: 0.5133
+Epoch 11/50 map_50: 0.5543
+Epoch 11/50 map_75: 0.3871
+Epoch 11/50 mar_1: 0.3144
+Epoch 11/50 mar_10: 0.5052
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_11.pt...
+Time cost: 217m36.1s
+---
+Epoch 12/50 training_loss: 7.7819
+Epoch 12/50 map: 0.3651
+Epoch 12/50 map_small: 0.0797
+Epoch 12/50 map_medium: 0.2550
+Epoch 12/50 map_large: 0.5191
+Epoch 12/50 map_50: 0.5583
+Epoch 12/50 map_75: 0.3904
+Epoch 12/50 mar_1: 0.3152
+Epoch 12/50 mar_10: 0.5105
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_12.pt...
+Time cost: 217m27.4s
+---
+Epoch 13/50 training_loss: 7.7014
+Epoch 13/50 map: 0.3667
+Epoch 13/50 map_small: 0.0829
+Epoch 13/50 map_medium: 0.2575
+Epoch 13/50 map_large: 0.5176
+Epoch 13/50 map_50: 0.5666
+Epoch 13/50 map_75: 0.3882
+Epoch 13/50 mar_1: 0.3160
+Epoch 13/50 mar_10: 0.5091
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_13.pt...
+Time cost: 217m20.2s
+---
+Epoch 14/50 training_loss: 7.6573
+Epoch 14/50 map: 0.3675
+Epoch 14/50 map_small: 0.0864
+Epoch 14/50 map_medium: 0.2563
+Epoch 14/50 map_large: 0.5284
+Epoch 14/50 map_50: 0.5642
+Epoch 14/50 map_75: 0.3906
+Epoch 14/50 mar_1: 0.3181
+Epoch 14/50 mar_10: 0.5111
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_14.pt...
+Time cost: 217m18.2s
+---
+Epoch 15/50 training_loss: 7.6031
+Epoch 15/50 map: 0.3794
+Epoch 15/50 map_small: 0.0893
+Epoch 15/50 map_medium: 0.2700
+Epoch 15/50 map_large: 0.5301
+Epoch 15/50 map_50: 0.5763
+Epoch 15/50 map_75: 0.4069
+Epoch 15/50 mar_1: 0.3216
+Epoch 15/50 mar_10: 0.5228
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_15.pt...
+Time cost: 217m55.4s
+---
+Epoch 16/50 training_loss: 7.5628
+Epoch 16/50 map: 0.3801
+Epoch 16/50 map_small: 0.0943
+Epoch 16/50 map_medium: 0.2690
+Epoch 16/50 map_large: 0.5346
+Epoch 16/50 map_50: 0.5734
+Epoch 16/50 map_75: 0.4098
+Epoch 16/50 mar_1: 0.3257
+Epoch 16/50 mar_10: 0.5271
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_16.pt...
+Time cost: 217m30.4s
+---
+Epoch 17/50 training_loss: 7.5019
+Epoch 17/50 map: 0.3779
+Epoch 17/50 map_small: 0.0919
+Epoch 17/50 map_medium: 0.2628
+Epoch 17/50 map_large: 0.5379
+Epoch 17/50 map_50: 0.5769
+Epoch 17/50 map_75: 0.4035
+Epoch 17/50 mar_1: 0.3238
+Epoch 17/50 mar_10: 0.5173
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_17.pt...
+Time cost: 217m06.6s
+---
+Epoch 18/50 training_loss: 7.4457
+Epoch 18/50 map: 0.3877
+Epoch 18/50 map_small: 0.0894
+Epoch 18/50 map_medium: 0.2721
+Epoch 18/50 map_large: 0.5426
+Epoch 18/50 map_50: 0.5853
+Epoch 18/50 map_75: 0.4147
+Epoch 18/50 mar_1: 0.3238
+Epoch 18/50 mar_10: 0.5222
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_18.pt...
+Time cost: 217m40.2s
+---
+Epoch 19/50 training_loss: 7.4265
+Epoch 19/50 map: 0.3877
+Epoch 19/50 map_small: 0.0884
+Epoch 19/50 map_medium: 0.2751
+Epoch 19/50 map_large: 0.5411
+Epoch 19/50 map_50: 0.5876
+Epoch 19/50 map_75: 0.4135
+Epoch 19/50 mar_1: 0.3262
+Epoch 19/50 mar_10: 0.5275
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_19.pt...
+Time cost: 217m49.7s
+---
+Epoch 20/50 training_loss: 7.4091
+Epoch 20/50 map: 0.3864
+Epoch 20/50 map_small: 0.0892
+Epoch 20/50 map_medium: 0.2743
+Epoch 20/50 map_large: 0.5481
+Epoch 20/50 map_50: 0.5879
+Epoch 20/50 map_75: 0.4102
+Epoch 20/50 mar_1: 0.3251
+Epoch 20/50 mar_10: 0.5261
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_20.pt...
+Time cost: 217m38.6s
+---
+Epoch 21/50 training_loss: 7.3360
+Epoch 21/50 map: 0.3909
+Epoch 21/50 map_small: 0.1011
+Epoch 21/50 map_medium: 0.2836
+Epoch 21/50 map_large: 0.5447
+Epoch 21/50 map_50: 0.5892
+Epoch 21/50 map_75: 0.4186
+Epoch 21/50 mar_1: 0.3290
+Epoch 21/50 mar_10: 0.5324
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_21.pt...
+Time cost: 217m27.9s
+---
+Epoch 22/50 training_loss: 7.3009
+Epoch 22/50 map: 0.3963
+Epoch 22/50 map_small: 0.0915
+Epoch 22/50 map_medium: 0.2869
+Epoch 22/50 map_large: 0.5479
+Epoch 22/50 map_50: 0.5985
+Epoch 22/50 map_75: 0.4249
+Epoch 22/50 mar_1: 0.3276
+Epoch 22/50 mar_10: 0.5302
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_22.pt...
+Time cost: 217m20.4s
+---
+Epoch 23/50 training_loss: 7.2577
+Epoch 23/50 map: 0.3896
+Epoch 23/50 map_small: 0.0942
+Epoch 23/50 map_medium: 0.2778
+Epoch 23/50 map_large: 0.5440
+Epoch 23/50 map_50: 0.5946
+Epoch 23/50 map_75: 0.4181
+Epoch 23/50 mar_1: 0.3246
+Epoch 23/50 mar_10: 0.5280
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_23.pt...
+Time cost: 217m27.9s
+---
+Epoch 24/50 training_loss: 7.2256
+Epoch 24/50 map: 0.3885
+Epoch 24/50 map_small: 0.0915
+Epoch 24/50 map_medium: 0.2758
+Epoch 24/50 map_large: 0.5501
+Epoch 24/50 map_50: 0.5901
+Epoch 24/50 map_75: 0.4131
+Epoch 24/50 mar_1: 0.3243
+Epoch 24/50 mar_10: 0.5301
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_24.pt...
+Time cost: 217m30.4s
+---
+Epoch 25/50 training_loss: 7.2125
+Epoch 25/50 map: 0.3900
+Epoch 25/50 map_small: 0.0881
+Epoch 25/50 map_medium: 0.2798
+Epoch 25/50 map_large: 0.5462
+Epoch 25/50 map_50: 0.5928
+Epoch 25/50 map_75: 0.4166
+Epoch 25/50 mar_1: 0.3294
+Epoch 25/50 mar_10: 0.5274
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_25.pt...
+Time cost: 217m56.8s
+---
+Epoch 26/50 training_loss: 7.2075
+Epoch 26/50 map: 0.3892
+Epoch 26/50 map_small: 0.0887
+Epoch 26/50 map_medium: 0.2841
+Epoch 26/50 map_large: 0.5438
+Epoch 26/50 map_50: 0.5925
+Epoch 26/50 map_75: 0.4156
+Epoch 26/50 mar_1: 0.3254
+Epoch 26/50 mar_10: 0.5283
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_26.pt...
+Time cost: 220m24.4s
+---
+Epoch 27/50 training_loss: 7.1721
+Epoch 27/50 map: 0.3906
+Epoch 27/50 map_small: 0.0924
+Epoch 27/50 map_medium: 0.2847
+Epoch 27/50 map_large: 0.5432
+Epoch 27/50 map_50: 0.5934
+Epoch 27/50 map_75: 0.4189
+Epoch 27/50 mar_1: 0.3266
+Epoch 27/50 mar_10: 0.5280
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_27.pt...
+Time cost: 222m05.7s
+---
+Epoch 28/50 training_loss: 7.1610
+Epoch 28/50 map: 0.4020
+Epoch 28/50 map_small: 0.0871
+Epoch 28/50 map_medium: 0.2913
+Epoch 28/50 map_large: 0.5555
+Epoch 28/50 map_50: 0.6008
+Epoch 28/50 map_75: 0.4297
+Epoch 28/50 mar_1: 0.3322
+Epoch 28/50 mar_10: 0.5352
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_28.pt...
+Time cost: 219m38.1s
+---
+Epoch 29/50 training_loss: 7.1040
+Epoch 29/50 map: 0.4012
+Epoch 29/50 map_small: 0.0981
+Epoch 29/50 map_medium: 0.2925
+Epoch 29/50 map_large: 0.5542
+Epoch 29/50 map_50: 0.6050
+Epoch 29/50 map_75: 0.4319
+Epoch 29/50 mar_1: 0.3321
+Epoch 29/50 mar_10: 0.5392
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_29.pt...
+Time cost: 217m11.8s
+---
+Epoch 30/50 training_loss: 7.0705
+Epoch 30/50 map: 0.3993
+Epoch 30/50 map_small: 0.0991
+Epoch 30/50 map_medium: 0.2903
+Epoch 30/50 map_large: 0.5529
+Epoch 30/50 map_50: 0.6063
+Epoch 30/50 map_75: 0.4270
+Epoch 30/50 mar_1: 0.3306
+Epoch 30/50 mar_10: 0.5388
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_30.pt...
+Time cost: 217m28.8s
+---
+Epoch 31/50 training_loss: 7.0703
+Epoch 31/50 map: 0.4013
+Epoch 31/50 map_small: 0.0964
+Epoch 31/50 map_medium: 0.2947
+Epoch 31/50 map_large: 0.5534
+Epoch 31/50 map_50: 0.6065
+Epoch 31/50 map_75: 0.4280
+Epoch 31/50 mar_1: 0.3321
+Epoch 31/50 mar_10: 0.5395
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_31.pt...
+Time cost: 217m19.6s
+---
+Epoch 32/50 training_loss: 7.0355
+Epoch 32/50 map: 0.3998
+Epoch 32/50 map_small: 0.0956
+Epoch 32/50 map_medium: 0.2894
+Epoch 32/50 map_large: 0.5548
+Epoch 32/50 map_50: 0.6053
+Epoch 32/50 map_75: 0.4280
+Epoch 32/50 mar_1: 0.3303
+Epoch 32/50 mar_10: 0.5358
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_32.pt...
+Time cost: 217m45.2s
+---
+Epoch 33/50 training_loss: 6.9822
+Epoch 33/50 map: 0.3990
+Epoch 33/50 map_small: 0.1013
+Epoch 33/50 map_medium: 0.2920
+Epoch 33/50 map_large: 0.5528
+Epoch 33/50 map_50: 0.6063
+Epoch 33/50 map_75: 0.4276
+Epoch 33/50 mar_1: 0.3277
+Epoch 33/50 mar_10: 0.5335
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_33.pt...
+Time cost: 217m16.2s
+---
+Epoch 34/50 training_loss: 6.9857
+Epoch 34/50 map: 0.4010
+Epoch 34/50 map_small: 0.0968
+Epoch 34/50 map_medium: 0.2915
+Epoch 34/50 map_large: 0.5568
+Epoch 34/50 map_50: 0.6082
+Epoch 34/50 map_75: 0.4276
+Epoch 34/50 mar_1: 0.3302
+Epoch 34/50 mar_10: 0.5308
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_34.pt...
+Time cost: 219m44.1s
+---
+Epoch 35/50 training_loss: 6.9680
+Epoch 35/50 map: 0.4066
+Epoch 35/50 map_small: 0.1003
+Epoch 35/50 map_medium: 0.2983
+Epoch 35/50 map_large: 0.5633
+Epoch 35/50 map_50: 0.6090
+Epoch 35/50 map_75: 0.4380
+Epoch 35/50 mar_1: 0.3332
+Epoch 35/50 mar_10: 0.5371
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_35.pt...
+Time cost: 222m38.4s
+---
+Epoch 36/50 training_loss: 6.9488
+Epoch 36/50 map: 0.3992
+Epoch 36/50 map_small: 0.0965
+Epoch 36/50 map_medium: 0.2916
+Epoch 36/50 map_large: 0.5598
+Epoch 36/50 map_50: 0.6049
+Epoch 36/50 map_75: 0.4253
+Epoch 36/50 mar_1: 0.3325
+Epoch 36/50 mar_10: 0.5413
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_36.pt...
+Time cost: 217m22.0s
+---
+Epoch 37/50 training_loss: 6.9647
+Epoch 37/50 map: 0.4039
+Epoch 37/50 map_small: 0.0954
+Epoch 37/50 map_medium: 0.2981
+Epoch 37/50 map_large: 0.5596
+Epoch 37/50 map_50: 0.6084
+Epoch 37/50 map_75: 0.4315
+Epoch 37/50 mar_1: 0.3328
+Epoch 37/50 mar_10: 0.5379
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_37.pt...
+Time cost: 219m51.8s
+---
+Epoch 38/50 training_loss: 6.9387
+Epoch 38/50 map: 0.4013
+Epoch 38/50 map_small: 0.0926
+Epoch 38/50 map_medium: 0.2953
+Epoch 38/50 map_large: 0.5580
+Epoch 38/50 map_50: 0.6103
+Epoch 38/50 map_75: 0.4292
+Epoch 38/50 mar_1: 0.3301
+Epoch 38/50 mar_10: 0.5382
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_38.pt...
+Time cost: 222m32.4s
+---
+Epoch 39/50 training_loss: 6.9389
+Epoch 39/50 map: 0.4064
+Epoch 39/50 map_small: 0.0933
+Epoch 39/50 map_medium: 0.2989
+Epoch 39/50 map_large: 0.5603
+Epoch 39/50 map_50: 0.6116
+Epoch 39/50 map_75: 0.4357
+Epoch 39/50 mar_1: 0.3365
+Epoch 39/50 mar_10: 0.5425
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_39.pt...
+Time cost: 221m57.5s
+---
+Epoch 40/50 training_loss: 6.9207
+Updated learning rate to: 2e-05
+Epoch 40/50 map: 0.4059
+Epoch 40/50 map_small: 0.0970
+Epoch 40/50 map_medium: 0.2961
+Epoch 40/50 map_large: 0.5641
+Epoch 40/50 map_50: 0.6109
+Epoch 40/50 map_75: 0.4348
+Epoch 40/50 mar_1: 0.3360
+Epoch 40/50 mar_10: 0.5421
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_40.pt...
+Time cost: 219m03.2s
+---
+Epoch 41/50 training_loss: 6.5086
+Epoch 41/50 map: 0.4253
+Epoch 41/50 map_small: 0.1086
+Epoch 41/50 map_medium: 0.3137
+Epoch 41/50 map_large: 0.5824
+Epoch 41/50 map_50: 0.6318
+Epoch 41/50 map_75: 0.4533
+Epoch 41/50 mar_1: 0.3428
+Epoch 41/50 mar_10: 0.5562
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_41.pt...
+Time cost: 220m35.1s
+---
+Epoch 42/50 training_loss: 6.3571
+Epoch 42/50 map: 0.4284
+Epoch 42/50 map_small: 0.1117
+Epoch 42/50 map_medium: 0.3184
+Epoch 42/50 map_large: 0.5855
+Epoch 42/50 map_50: 0.6369
+Epoch 42/50 map_75: 0.4562
+Epoch 42/50 mar_1: 0.3443
+Epoch 42/50 mar_10: 0.5591
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_42.pt...
+Time cost: 218m34.4s
+---
+Epoch 43/50 training_loss: 6.2926
+Epoch 43/50 map: 0.4309
+Epoch 43/50 map_small: 0.1148
+Epoch 43/50 map_medium: 0.3220
+Epoch 43/50 map_large: 0.5882
+Epoch 43/50 map_50: 0.6382
+Epoch 43/50 map_75: 0.4595
+Epoch 43/50 mar_1: 0.3479
+Epoch 43/50 mar_10: 0.5643
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_43.pt...
+Time cost: 218m15.5s
+---
+Epoch 44/50 training_loss: 6.2471
+Epoch 44/50 map: 0.4294
+Epoch 44/50 map_small: 0.1139
+Epoch 44/50 map_medium: 0.3206
+Epoch 44/50 map_large: 0.5867
+Epoch 44/50 map_50: 0.6367
+Epoch 44/50 map_75: 0.4579
+Epoch 44/50 mar_1: 0.3458
+Epoch 44/50 mar_10: 0.5631
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_44.pt...
+Time cost: 218m03.8s
+---
+Epoch 45/50 training_loss: 6.2105
+Epoch 45/50 map: 0.4308
+Epoch 45/50 map_small: 0.1128
+Epoch 45/50 map_medium: 0.3212
+Epoch 45/50 map_large: 0.5901
+Epoch 45/50 map_50: 0.6378
+Epoch 45/50 map_75: 0.4607
+Epoch 45/50 mar_1: 0.3460
+Epoch 45/50 mar_10: 0.5646
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_45.pt...
+Time cost: 217m47.4s
+---
+Epoch 46/50 training_loss: 6.1820
+Epoch 46/50 map: 0.4336
+Epoch 46/50 map_small: 0.1156
+Epoch 46/50 map_medium: 0.3260
+Epoch 46/50 map_large: 0.5919
+Epoch 46/50 map_50: 0.6409
+Epoch 46/50 map_75: 0.4617
+Epoch 46/50 mar_1: 0.3475
+Epoch 46/50 mar_10: 0.5666
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_46.pt...
+Time cost: 218m37.4s
+---
+Epoch 47/50 training_loss: 6.1621
+Epoch 47/50 map: 0.4339
+Epoch 47/50 map_small: 0.1172
+Epoch 47/50 map_medium: 0.3246
+Epoch 47/50 map_large: 0.5927
+Epoch 47/50 map_50: 0.6416
+Epoch 47/50 map_75: 0.4628
+Epoch 47/50 mar_1: 0.3484
+Epoch 47/50 mar_10: 0.5655
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_47.pt...
+Time cost: 218m49.0s
+---
+Epoch 48/50 training_loss: 6.1525
+Epoch 48/50 map: 0.4349
+Epoch 48/50 map_small: 0.1158
+Epoch 48/50 map_medium: 0.3269
+Epoch 48/50 map_large: 0.5941
+Epoch 48/50 map_50: 0.6435
+Epoch 48/50 map_75: 0.4645
+Epoch 48/50 mar_1: 0.3474
+Epoch 48/50 mar_10: 0.5675
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_48.pt...
+Time cost: 217m35.4s
+---
+Epoch 49/50 training_loss: 6.1166
+Epoch 49/50 map: 0.4356
+Epoch 49/50 map_small: 0.1173
+Epoch 49/50 map_medium: 0.3272
+Epoch 49/50 map_large: 0.5942
+Epoch 49/50 map_50: 0.6438
+Epoch 49/50 map_75: 0.4672
+Epoch 49/50 mar_1: 0.3490
+Epoch 49/50 mar_10: 0.5678
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_49.pt...
+Time cost: 217m41.9s
+---
+Epoch 50/60 training_loss: 6.0881
+Epoch 50/60 map: 0.4373
+Epoch 50/60 map_small: 0.1163
+Epoch 50/60 map_medium: 0.3256
+Epoch 50/60 map_large: 0.5958
+Epoch 50/60 map_50: 0.6453
+Epoch 50/60 map_75: 0.4672
+Epoch 50/60 mar_1: 0.3490
+Epoch 50/60 mar_10: 0.5678
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_50.pt...
+Time cost: 218m05.1s
+---
+Saving model checkpoint models/deformable_detr_boxref_coco_convnext_v2_tiny_imagenet21k_50.pt...

training_args.json

@@ -0,0 +1,98 @@
+{
+  "cmdline": "train_detection.py --network deformable_detr_boxref --tag coco --backbone convnext_v2_tiny --backbone-tag imagenet21k --backbone-pretrained --opt adamw --lr 0.0002 --backbone-lr 0.00002 --lr-scheduler step --lr-step-size 40 --lr-step-gamma 0.1 --batch-size 2 --epochs 50 --wd 0.0001 --grad-accum-steps 16 --multiscale --max-size 1152 --clip-grad-norm 0.1 --amp --amp-dtype bfloat16 --compile-backbone --compile-opt --save-frequency 1 --data-path ~/Datasets/cocodataset/train2017 --val-path ~/Datasets/cocodataset/val2017 --coco-json-path ~/Datasets/cocodataset/annotations/instances_train2017.json --coco-val-json-path ~/Datasets/cocodataset/annotations/instances_val2017.json --class-file public_datasets_metadata/coco-classes.txt --no-summary",
+  "network": "deformable_detr_boxref",
+  "net_param": null,
+  "model_config": null,
+  "backbone": "convnext_v2_tiny",
+  "backbone_param": null,
+  "backbone_model_config": null,
+  "backbone_tag": "imagenet21k",
+  "backbone_epoch": null,
+  "backbone_pretrained": true,
+  "pretrained": false,
+  "reset_head": false,
+  "freeze_body": false,
+  "freeze_backbone": false,
+  "freeze_backbone_stages": null,
+  "compile": false,
+  "compile_backbone": true,
+  "compile_dynamic": false,
+  "compile_opt": true,
+  "opt": "adamw",
+  "momentum": 0.9,
+  "nesterov": false,
+  "opt_eps": null,
+  "opt_betas": null,
+  "opt_alpha": null,
+  "lr": 0.0002,
+  "bias_lr": null,
+  "backbone_lr": 2e-05,
+  "lr_scale": null,
+  "lr_scale_type": "linear",
+  "wd": 0.0001,
+  "norm_wd": null,
+  "bias_weight_decay": null,
+  "transformer_embedding_decay": null,
+  "layer_decay": null,
+  "lr_scheduler_update": "epoch",
+  "lr_scheduler": "step",
+  "lr_step_size": 40,
+  "lr_steps": null,
+  "lr_step_gamma": 0.1,
+  "lr_cosine_min": 1e-06,
+  "lr_power": 1.0,
+  "grad_accum_steps": 16,
+  "channels": 3,
+  "size": [
+    640,
+    640
+  ],
+  "max_size": 1152,
+  "dynamic_size": false,
+  "multiscale": true,
+  "freeze_backbone_bn": false,
+  "sync_bn": false,
+  "batch_size": 2,
+  "warmup_epochs": 0,
+  "aug_type": "birder",
+  "aug_level": 4,
+  "rgb_mode": "birder",
+  "epochs": 50,
+  "stop_epoch": 51,
+  "save_frequency": 1,
+  "keep_last": null,
+  "resume_epoch": null,
+  "load_states": false,
+  "load_scheduler": false,
+  "model_ema": false,
+  "model_ema_steps": 1,
+  "model_ema_decay": 0.9998,
+  "ra_sampler": false,
+  "ra_reps": 3,
+  "tag": "coco",
+  "log_interval": 20,
+  "num_workers": 8,
+  "prefetch_factor": null,
+  "amp": true,
+  "amp_dtype": "bfloat16",
+  "fast_matmul": false,
+  "grad_anomaly_detection": false,
+  "world_size": 2,
+  "dist_url": "env://",
+  "clip_grad_norm": 0.1,
+  "gpu": 0,
+  "cpu": false,
+  "use_deterministic_algorithms": false,
+  "binary_mode": false,
+  "plot_lr": false,
+  "no_summary": true,
+  "val_path": "~/Datasets/cocodataset/val2017",
+  "data_path": "~/Datasets/cocodataset/train2017",
+  "coco_val_json_path": "~/Datasets/cocodataset/annotations/instances_val2017.json",
+  "coco_json_path": "~/Datasets/cocodataset/annotations/instances_train2017.json",
+  "class_file": "public_datasets_metadata/coco-classes.txt",
+  "ignore_file": null,
+  "rank": 0,
+  "distributed": true,
+  "dist_backend": "nccl"
+}