facebookresearch / dino

PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
Apache License 2.0
6.25k stars 905 forks source link

How to fine-tune on a custom dataset? #193

Open rringham opened 2 years ago

rringham commented 2 years ago

Is there a guide for how to fine-tune DINO on a custom imagenet-formatted dataset? (after pretraining on custom data)

Harry-KIT commented 2 years ago

Hi @rringham. Please check it #80