Useful callbacks for self-supervised learning models.
Note
We rely on the community to keep these updated and working. If something doesn't work, we'd really appreciate a contribution to fix!
The exponential moving average weight-update rule from Bootstrap Your Own Latent (BYOL).
pl_bolts.callbacks.byol_updates.BYOLMAWeightUpdate
Appends a MLP for fine-tuning to the given model. Callback has its own mini-inner loop.
pl_bolts.callbacks.ssl_online.SSLOnlineEvaluator