lucidrains / contrastive-learner
A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network
☆135Updated 4 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☆100Updated 2 years ago
- Code to train the SimSiam model on cifar10 using PyTorch☆40Updated 3 years ago
- The official repository for "Intermediate Layers Matter in Momentum Contrastive Self Supervised Learning" paper.☆40Updated 2 years ago
- Official PyTorch implementation of the paper "Self-Supervised Relational Reasoning for Representation Learning", NeurIPS 2020 Spotlight.☆142Updated 11 months ago
- MODALS: Modality-agnostic Automated Data Augmentation in the Latent Space☆40Updated 4 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆131Updated 3 weeks ago
- ☆58Updated 3 years ago
- ☆92Updated 4 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆259Updated 4 years ago
- NeurIPS 2020, Debiased Contrastive Learning☆283Updated 2 years ago
- An essential implementation of BYOL in PyTorch + PyTorch Lightning☆51Updated 3 years ago
- Pytorch implementation of "A Simple Framework for Contrastive Learning of Visual Representations"☆82Updated last year
- Unofficial Pytorch code for "FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence" in NeurIPS'20. This repo co…☆190Updated 4 years ago
- An education step by step implementation of SimCLR that accompanies the blogpost☆32Updated 3 years ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 4 years ago
- A Domain-Agnostic Benchmark for Self-Supervised Learning☆107Updated last year
- Pretrained SimCLRv2 models in Pytorch☆105Updated 4 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆116Updated 4 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆220Updated 2 years ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆203Updated 11 months ago
- Official source code repository for the ICML 2021 paper "Hierarchical VAEs Know What They Don't Know"☆30Updated 3 years ago
- [ICML 2020] code for the flooding regularizer proposed in "Do We Need Zero Training Loss After Achieving Zero Training Error?"☆91Updated 2 years ago
- TensorFlow implementation of "Unsupervised Learning of Visual Features by Contrasting Cluster Assignments".☆86Updated 2 years ago
- Implementation for our WACV 2021 paper "Multi-Loss Weighting with Coefficient of Variations"☆50Updated 4 years ago
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆77Updated 2 years ago
- Pytorch implementations of generative models: VQVAE2, AIR, DRAW, InfoGAN, DCGAN, SSVAE☆91Updated 4 years ago
- Axial Positional Embedding for Pytorch☆77Updated last month
- Official Pytorch implementation of ReBias (Learning De-biased Representations with Biased Representations), ICML 2020☆173Updated 8 months ago
- Implementation of the paper Self-Supervised Learning of Pretext-Invariant Representations☆89Updated 4 years ago