Load DETR's object detection model and specify your hyperparameters for the model's training.

detr_model.py

 # Training the DETR model
model = AutoModelForObjectDetection.from_pretrained(
    checkpoint,
    id2label=id2label,
    label2id=label2id,
    ignore_mismatched_sizes=True,
)

training_args = TrainingArguments(
    output_dir="detr-resnet-50_finetuned_cppe5",
    # overwrite_output_dir= ,
    per_device_train_batch_size=8,
    num_train_epochs=40,
    fp16=False,
    save_steps=200,
    logging_steps=10,
    learning_rate=1e-5,
    weight_decay=1e-4,
    save_total_limit=2,
    remove_unused_columns=False,
    push_to_hub=False,
)