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☆99Updated 2 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆133Updated last month
- The official repository for "Intermediate Layers Matter in Momentum Contrastive Self Supervised Learning" paper.☆40Updated 3 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆130Updated 2 years ago
- ☆58Updated 3 years ago
- MODALS: Modality-agnostic Automated Data Augmentation in the Latent Space☆40Updated 4 years ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆203Updated last year
- Code to train the SimSiam model on cifar10 using PyTorch☆40Updated 3 years ago
- Pytorch implementation of "A Simple Framework for Contrastive Learning of Visual Representations"☆82Updated last year
- [ICML 2020] code for the flooding regularizer proposed in "Do We Need Zero Training Loss After Achieving Zero Training Error?"☆91Updated 2 years ago
- Official PyTorch implementation of the paper "Self-Supervised Relational Reasoning for Representation Learning", NeurIPS 2020 Spotlight.☆142Updated last year
- NeurIPS 2020, Debiased Contrastive Learning☆282Updated 2 years ago
- An essential implementation of BYOL in PyTorch + PyTorch Lightning☆51Updated 3 years ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆258Updated 4 years ago
- Pretrained SimCLRv2 models in Pytorch☆105Updated 4 years ago
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆77Updated 2 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆220Updated 2 years ago
- Unofficial Pytorch code for "FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence" in NeurIPS'20. This repo co…☆190Updated 4 years ago
- A Domain-Agnostic Benchmark for Self-Supervised Learning☆107Updated last year
- A Pytorch implementation of the optimal transport kernel embedding☆116Updated 4 years ago
- Axial Positional Embedding for Pytorch☆79Updated 2 months ago
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆160Updated last year
- ☆92Updated 4 years ago
- Implementation for our WACV 2021 paper "Multi-Loss Weighting with Coefficient of Variations"☆50Updated 4 years ago
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆69Updated 4 years ago
- ☆48Updated 2 years ago
- A PyTorch reproduction of 'A Simple Framework for Contrastive Learning of Visual Representations' by Ting Chen, et al.☆48Updated 10 months ago
- An education step by step implementation of SimCLR that accompanies the blogpost☆32Updated 3 years ago
- Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.☆143Updated 3 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago