Register the loss center for proper DINO checkpoint resuming

facebookresearch / vissl

VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.

https://vissl.ai

MIT License

3.26k stars 334 forks source link

Closed prigoyal closed 3 years ago

prigoyal commented 3 years ago

@imisra pointed out this issue and fixing it.

facebook-github-bot commented 3 years ago

@prigoyal has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.