center checkpint - Githubissues

facebookresearch / dino

PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO

Apache License 2.0

6.38k stars 910 forks source link

center checkpint #188

Closed fawazsammani closed 2 years ago

fawazsammani commented 2 years ago

Hi! Thanks for your work. Would it be possible to share the "center" buffer used to store the exp moving average of the teacher mean (the DINOLoss state_dic) for the Reset-50 model? The full_chekpoint you provided only has the teacher and student state_dic Hope to get a reply soon and thanks!