Skip to content

davidlightmysterion/detr-huggingface

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

detr-huggingface

This repository is a reimplementation of the facebook torch DETR project using the Hugging Face Trainer.

Results

  • Trained on 4x A100 80GB GPUs.

  • Due to an unexpected issue, training stopped at 137 epochs.

  • Achieved 37.8 mAP, slightly below the official implementation (39.4 mAP).

eval_map.png

train_loss.png

About

a reimplementation of the facebook torch DETR project using Huggingface Trainer

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages