Skip to content

Custom Backbones Made Easy

Compare
Choose a tag to compare
@philippmwirth philippmwirth released this 10 Dec 15:15
· 1113 commits to master since this release
1a5bc35

Custom Backbones Made Easy

Decoupling Self-Supervised Models from ResNet

The implementation of SimCLR and MoCo have been changed such that they can now be constructed from an arbitrary backbone network.
Furthermore, the backbone of the self-supervised models is now called backbone instead of features.

Big Documentation Update

The documentation has received a lot of love and improvements to make working with lightly easier.
There is also a new tutorial on how to train MoCo on Cifar-10.

Minor Changes

The LightlyDataset can now be passed a list of indices marking relevant samples. Non-relevant samples will be ignored during further processing of the data.

Models