Store center buffer in DINO training

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.25k stars 331 forks source link

Closed prigoyal closed 2 years ago

prigoyal commented 2 years ago

Summary: Correctly store the buffer and check the training resumes fine when interrupted.

Differential Revision: D31395224

facebook-github-bot commented 2 years ago

This pull request was exported from Phabricator. Differential Revision: D31395224