lucidrains / contrastive-learner
A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network
☆132Updated 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
- The official repository for "Intermediate Layers Matter in Momentum Contrastive Self Supervised Learning" paper.☆40Updated 2 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆129Updated last year
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- NeurIPS 2020, Debiased Contrastive Learning☆283Updated 2 years ago
- ☆58Updated 3 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
- Code to train the SimSiam model on cifar10 using PyTorch☆40Updated 3 years ago
- Official implementation of Auxiliary Learning by Implicit Differentiation [ICLR 2021]☆84Updated 7 months ago
- Experiments with supervised contrastive learning methods with different loss functions☆219Updated 2 years ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 4 years ago
- An essential implementation of BYOL in PyTorch + PyTorch Lightning☆51Updated 3 years ago
- ☆92Updated 4 years ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆202Updated 10 months ago
- A PyTorch reproduction of 'A Simple Framework for Contrastive Learning of Visual Representations' by Ting Chen, et al.☆47Updated 9 months ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆258Updated 4 years ago
- Code for the paper "Calibrating Deep Neural Networks using Focal Loss"☆160Updated last year
- Pytorch implementation of "A Simple Framework for Contrastive Learning of Visual Representations"☆81Updated 11 months ago
- Implementation of Feedback Transformer in Pytorch☆105Updated 4 years ago
- [NeurIPS 2021] Official codes for "Efficient Training of Visual Transformers with Small Datasets".☆141Updated 2 months ago
- Estimating Example Difficulty using Variance of Gradients☆63Updated 2 years ago
- Pretrained SimCLRv2 models in Pytorch☆105Updated 4 years ago
- pytorch implementation of manifold-mixup☆22Updated 2 years ago
- Awesome Contrastive Learning for CV & NLP☆162Updated 3 years ago
- Figures & code from the paper "Shortcut Learning in Deep Neural Networks" (Nature Machine Intelligence 2020)☆96Updated 2 years ago
- A PyTorch converter for SimCLR checkpoints☆109Updated 4 years ago
- EsViT: Efficient self-supervised Vision Transformers☆410Updated last year
- Codebase for Learning Invariances in Neural Networks☆94Updated 2 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆114Updated 4 years ago