lucidrains / contrastive-learner
A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network
☆129Updated 3 years ago
Alternatives and similar repositories for contrastive-learner:
Users that are interested in contrastive-learner are comparing it to the libraries listed below
- Implementation of Barlow Twins paper☆99Updated 2 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆124Updated last year
- Official PyTorch implementation of the paper "Self-Supervised Relational Reasoning for Representation Learning", NeurIPS 2020 Spotlight.☆142Updated 9 months ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆257Updated 4 years ago
- Code to train the SimSiam model on cifar10 using PyTorch☆39Updated 3 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- The official repository for "Intermediate Layers Matter in Momentum Contrastive Self Supervised Learning" paper.☆40Updated 2 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆218Updated 2 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- MODALS: Modality-agnostic Automated Data Augmentation in the Latent Space☆40Updated 4 years ago
- NeurIPS 2020, Debiased Contrastive Learning☆282Updated last year
- An essential implementation of BYOL in PyTorch + PyTorch Lightning☆51Updated 3 years ago
- A Domain-Agnostic Benchmark for Self-Supervised Learning☆107Updated last year
- ☆229Updated 3 years ago
- Supplementary code for the paper "Hyperbolic Image Embeddings".☆372Updated 3 years ago
- ☆58Updated 3 years ago
- VICRegL official code base☆223Updated last year
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆76Updated 2 years ago
- Implementation of Feedback Transformer in Pytorch☆105Updated 3 years ago
- Axial Positional Embedding for Pytorch☆69Updated last week
- [ICML 2020] code for the flooding regularizer proposed in "Do We Need Zero Training Loss After Achieving Zero Training Error?"☆92Updated 2 years ago
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆159Updated last year
- ☆92Updated 4 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆114Updated 4 years ago
- Pytorch implementation of "A Simple Framework for Contrastive Learning of Visual Representations"☆81Updated 10 months ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆198Updated 9 months ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 4 years ago
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆193Updated 3 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆204Updated last year
- A PyTorch reproduction of 'A Simple Framework for Contrastive Learning of Visual Representations' by Ting Chen, et al.☆47Updated 8 months ago