Skip to content

MoCo and New Documentation

Compare
Choose a tag to compare
@philippmwirth philippmwirth released this 30 Oct 11:01
· 1225 commits to master since this release
86e51f0

MoCo and New Documentation

New Model: MoCo

lightly.models.moco.ResNetMoCo implements the momentum encoder architecture for self-supervised visual representation learning.
lightly.loss.memory_bank.MemoryBankWrapper allows the training of self-supervised models with a queue of negative samples.

New Documentation

The URL for the documentation has changed to https://docs.lightly.ai.
New section on how lightly works.
New tutorials have been added, check out the Pizza Tutorial to learn how to train a pizza classifier.

Further Changes:

Refactoring of lightly.api.
Default collate functions which implement the SimCLR and MoCo (v1) transfomations.
Collate functions work with tuple as input_size.
New tests and tox environments.
Removed sklearn dependency for PCA.

Models: