update json

config.json

@@ -1,5 +1,5 @@
 {
-    "model_type": "dfine",
+    "model_type": "d_fine",
     "architectures": ["DFineModel"],
     "auto_map": {
       "AutoConfig": "Laudando-Associates-LLC/d-fine-small--configuration_dfine.DFineConfig",

config.yaml

@@ -1,41 +0,0 @@
-task: detection
-
-evaluator:
-  type: CocoEvaluator
-  iou_types: ['bbox', ]
-
-num_classes: 3 # your dataset classes
-remap_mscoco_category: False
-
-train_dataloader:
-  type: DataLoader
-  dataset:
-    type: CocoDetection
-    img_folder: /dataset/images/train
-    ann_file: /dataset/annotations/instances_train.json
-    return_masks: False
-    transforms:
-      type: Compose
-      ops: ~
-  shuffle: True
-  num_workers: 4
-  drop_last: True
-  collate_fn:
-    type: BatchImageCollateFunction
-
-
-val_dataloader:
-  type: DataLoader
-  dataset:
-    type: CocoDetection
-    img_folder: /dataset/images/val
-    ann_file: /dataset/annotations/instances_val.json
-    return_masks: False
-    transforms:
-      type: Compose
-      ops: ~
-  shuffle: False
-  num_workers: 4
-  drop_last: False
-  collate_fn:
-    type: BatchImageCollateFunction

configuration_dfine.py

@@ -1,7 +1,7 @@
 from transformers import PretrainedConfig
 
 class DFineConfig(PretrainedConfig):
-    model_type = "dfine"
+    model_type = "d_fine"
 
     def __init__(
         self,