anzeyimana / Pytorch-PCGrad-GradVac-AMP-GradAccumLinks
PyTorch 1.11 reimplementation of multi task gradient adaptation ideas: Gradient Surgery (PCGrad) and Gradient Vaccine
☆21Updated 3 years ago
Alternatives and similar repositories for Pytorch-PCGrad-GradVac-AMP-GradAccum
Users that are interested in Pytorch-PCGrad-GradVac-AMP-GradAccum are comparing it to the libraries listed below
Sorting:
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆394Updated 4 years ago
- Official implementation of "Multi-Task Learning as a Bargaining Game" [ICML 2022]☆237Updated 7 months ago
- PyTorch custom dataset APIs -- CUB-200-2011, Stanford Dogs, Stanford Cars, FGVC Aircraft, NABirds, Tiny ImageNet, iNaturalist2017☆161Updated 3 years ago
- [ICLR 2021 Spotlight] Code release for "Long-tailed Recognition by Routing Diverse Distribution-Aware Experts."☆273Updated 2 years ago
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆144Updated 2 years ago
- Automated Search for Resource-Efficient Branched Multi-Task Networks [BMVC 2020]☆15Updated 2 years ago
- (Pytorch) Training ResNets on ImageNet-100 data☆64Updated 3 years ago
- Parametric Contrastive Learning (ICCV2021) & GPaCo (TPAMI 2023)☆259Updated 6 months ago
- PyTorch implementation of "Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning" with DDP and Apex AMP☆84Updated 5 years ago
- Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"☆433Updated 2 years ago
- Reproduce CKA: Similarity of Neural Network Representations Revisited☆311Updated 5 years ago
- Co^2L: Contrastive Continual Learning (ICCV2021)☆99Updated 3 years ago
- ☆128Updated 5 years ago
- Pytorch implementation of the paper Exploring Simple Siamese Representation Learning.☆76Updated 5 years ago
- [CVPR 2022 Oral] Crafting Better Contrastive Views for Siamese Representation Learning☆289Updated 3 years ago
- Pretrained SimCLRv2 models in Pytorch☆105Updated 5 years ago
- ImageNet-R(endition) and DeepAugment (ICCV 2021)☆281Updated 4 years ago
- An efficient implicit semantic augmentation method, complementary to existing non-semantic techniques.☆592Updated 4 years ago
- ☆429Updated 4 years ago
- Domain Generalization with MixStyle (ICLR'21)☆317Updated 3 years ago
- Official code for our CVPR '22 paper "Dataset Distillation by Matching Training Trajectories"☆436Updated last year
- Pytorch implementation for t-SNE with cuda to accelerate☆341Updated 2 years ago
- Code for "Gradient Surgery for Multi-Task Learning"☆345Updated 5 years ago
- The code repository for "A Model or 603 Exemplars: Towards Memory-Efficient Class-Incremental Learning" (ICLR'23) in PyTorch☆46Updated 2 years ago
- PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"☆611Updated 9 months ago
- Learning to Prompt (L2P) for Continual Learning @ CVPR22 and DualPrompt: Complementary Prompting for Rehearsal-free Continual Learning @ …☆473Updated last year
- PyTorch implementation of PiCO https://arxiv.org/abs/2201.08984☆225Updated 2 years ago
- A PyTorch implementation of SimCLR based on ICML 2020 paper "A Simple Framework for Contrastive Learning of Visual Representations"☆570Updated 5 years ago
- This is a PyTorch implementation of “Context AutoEncoder for Self-Supervised Representation Learning"☆198Updated 3 years ago
- This repository is the official Pytorch implementation of Self-Supervised Aggregation of Diverse Experts for Test-Agnostic Long-Tailed Re…☆166Updated 3 years ago