chrisby / torchMTL
A lightweight module for Multi-Task Learning in pytorch.
☆157Updated last year
Related projects ⓘ
Alternatives and complementary repositories for torchMTL
- ☆144Updated 2 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆784Updated 8 months ago
- Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.☆148Updated 5 years ago
- Metrics for model evaluation in pytorch☆110Updated 3 years ago
- A list of multi-task learning papers and projects.☆362Updated 2 years ago
- This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow☆134Updated 6 years ago
- Learning Rate Warmup in PyTorch☆393Updated last week
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆250Updated 2 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆251Updated 3 years ago
- ☆96Updated last year
- PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).☆778Updated 2 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.☆224Updated 9 months ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆169Updated 3 years ago
- Simple pytorch implementation of focal loss☆83Updated last year
- Official Pytorch Implementation of: "Asymmetric Loss For Multi-Label Classification"(ICCV, 2021) paper☆732Updated last year
- [NeurIPS 2020] Semi-Supervision (Unlabeled Data) & Self-Supervision Improve Class-Imbalanced / Long-Tailed Learning☆736Updated 3 years ago
- [ICML 2021, Long Talk] Delving into Deep Imbalanced Regression☆821Updated 2 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆417Updated 6 months ago
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆333Updated 2 years ago
- Class-Balanced Loss Based on Effective Number of Samples. CVPR 2019☆601Updated 3 years ago
- ☆51Updated 4 years ago
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆547Updated 3 years ago
- TensorFlow implementation of focal loss☆186Updated 3 years ago
- PyTorch samplers that output roughly balanced batches with support for multilabel datasets☆52Updated 7 months ago
- Unofficial PyTorch implementation of Fastformer based on paper "Fastformer: Additive Attention Can Be All You Need"."☆134Updated 3 years ago
- [NeurIPS 2019] Learning Imbalanced Datasets with Label-Distribution-Aware Margin Loss☆647Updated 2 years ago
- Ranger deep learning optimizer rewrite to use newest components☆322Updated 8 months ago
- Experiments with supervised contrastive learning methods with different loss functions☆214Updated last year
- Loss and accuracy go opposite ways...right?☆90Updated 4 years ago