lucidrains / contrastive-learner
A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network
☆128Updated 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 Nyström Self-attention, from the paper Nyströmformer☆124Updated 11 months ago
- The official repository for "Intermediate Layers Matter in Momentum Contrastive Self Supervised Learning" paper.☆40Updated 2 years ago
- MODALS: Modality-agnostic Automated Data Augmentation in the Latent Space☆40Updated 3 years ago
- Implementation of Barlow Twins paper☆99Updated 2 years ago
- NeurIPS 2020, Debiased Contrastive Learning☆282Updated last year
- ☆58Updated 3 years ago
- Official PyTorch implementation of the paper "Self-Supervised Relational Reasoning for Representation Learning", NeurIPS 2020 Spotlight.☆142Updated 8 months ago
- A Pytorch implementation of the optimal transport kernel embedding☆114Updated 3 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆130Updated last year
- A Domain-Agnostic Benchmark for Self-Supervised Learning☆107Updated last year
- Pytorch implementation of "A Simple Framework for Contrastive Learning of Visual Representations"☆80Updated 9 months ago
- Code for Neural Manifold Clustering and Embedding☆57Updated 2 years ago
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆68Updated 4 years ago
- ☆47Updated 2 years ago
- Code to train the SimSiam model on cifar10 using PyTorch☆39Updated 3 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated 2 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
- Code for the paper "Calibrating Deep Neural Networks using Focal Loss"☆158Updated last year
- Experiments with supervised contrastive learning methods with different loss functions☆218Updated 2 years ago
- TensorFlow implementation of "Unsupervised Learning of Visual Features by Contrasting Cluster Assignments".☆85Updated 2 years ago
- An essential implementation of BYOL in PyTorch + PyTorch Lightning☆51Updated 3 years ago
- Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆130Updated 2 years ago
- ☆92Updated 4 years ago
- [ICML 2020] code for the flooding regularizer proposed in "Do We Need Zero Training Loss After Achieving Zero Training Error?"☆92Updated 2 years ago
- [ICLR 2021] Concept Learners for Few-Shot Learning☆114Updated last year
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆258Updated 3 years ago
- Axial Positional Embedding for Pytorch☆66Updated last week
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆76Updated 2 years ago
- ☆226Updated 3 years ago
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆157Updated 11 months ago