# RF-DETR Medium — example training configuration (resolution 576, patch 16).
# Usage:
#   rfdetr fit --config configs/rfdetr_medium.yaml
#   rfdetr fit --config configs/rfdetr_medium.yaml \
#     --model.train_config.init_args.dataset_dir /data/my_dataset \
#     --trainer.devices 4

model:
  model_config:
    class_path: rfdetr.config.RFDETRMediumConfig
    init_args:
      num_classes: 80 # set to your dataset class count
  train_config:
    class_path: rfdetr.config.TrainConfig
    init_args:
      dataset_dir: /data/coco # required: path to your COCO-format dataset
      output_dir: output/rfdetr_medium
      epochs: 100
      batch_size: 4
      num_workers: 4
      tensorboard: true