LARS exclude BN: use a flag instead of a function

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 jingli9111 closed 3 years ago

jingli9111 commented 3 years ago

This fix the problem that the Barlow Twins model needs to save a function in the checkpoint.

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.

facebook-github-bot commented 3 years ago

@prigoyal merged this pull request in facebookresearch/vissl@43f230cd05a700426e21b7b79cb018d97198f370.