Image Classification
deit / README.md
lingkai's picture
Update README.md
751dc53 verified
metadata
license: mit
datasets:
  - ILSVRC/imagenet-1k
metrics:
  - accuracy
pipeline_tag: image-classification

Data efficient Image Transformers.

DeiT small model pretrained on imagenet with 300 epochs:

  • Acc@1 79.250 Acc@5 94.604 loss 0.927 *.

DeiT small model finetuned on imagenet with 15 epochs:

  • Acc@1 81.606 Acc@5 95.848 loss 0.80915 *.