lucidrains / contrastive-learnerLinks
A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network
☆139Updated 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
Sorting:
- Implementation of Barlow Twins paper☆99Updated 2 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆135Updated 2 months ago
- Official PyTorch implementation of the paper "Self-Supervised Relational Reasoning for Representation Learning", NeurIPS 2020 Spotlight.☆143Updated last year
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 2 years ago
- MODALS: Modality-agnostic Automated Data Augmentation in the Latent Space☆41Updated 4 years ago
- Axial Positional Embedding for Pytorch☆81Updated 3 months ago
- The official repository for "Intermediate Layers Matter in Momentum Contrastive Self Supervised Learning" paper.☆40Updated 3 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☆282Updated 2 years ago
- ☆93Updated 4 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆220Updated 2 years ago
- Implementation of Supervised Contrastive Learning with AMP, EMA, SWA, and many other tricks☆169Updated 2 years ago
- ☆58Updated 3 years ago
- 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
- 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…☆194Updated 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
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆205Updated last year
- Implementation of Feedback Transformer in Pytorch☆107Updated 4 years ago
- Code for the paper "Calibrating Deep Neural Networks using Focal Loss"☆161Updated last year
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆160Updated last year
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆77Updated 2 years ago
- Official Pytorch code for (AAAI 2020) paper "Capsule Routing via Variational Bayes", https://arxiv.org/pdf/1905.11455.pdf☆103Updated 3 years ago
- Pretrained SimCLRv2 models in Pytorch☆105Updated 4 years ago
- A Domain-Agnostic Benchmark for Self-Supervised Learning☆106Updated 2 years ago
- An essential implementation of BYOL in PyTorch + PyTorch Lightning☆51Updated 3 years ago
- EsViT: Efficient self-supervised Vision Transformers☆412Updated 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
- TensorFlow implementation of "Unsupervised Learning of Visual Features by Contrasting Cluster Assignments".☆86Updated 2 years ago