chrisby / torchMTL
A lightweight module for Multi-Task Learning in pytorch.
☆160Updated last year
Alternatives and similar repositories for torchMTL:
Users that are interested in torchMTL are comparing it to the libraries listed below
- ☆145Updated 3 years ago
- Metrics for model evaluation in pytorch☆113Updated 4 years ago
- A list of multi-task learning papers and projects.☆370Updated 3 years ago
- Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.☆149Updated 5 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆174Updated 3 years ago
- PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).☆801Updated 3 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆792Updated last year
- An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.☆233Updated last year
- A pytorch dataset sampler for always sampling balanced batches.☆114Updated 4 years ago
- PyTorch conversion of https://omoindrot.github.io/triplet-loss☆212Updated last year
- This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow☆136Updated 6 years ago
- Experimenting with different regression losses. Implemented in Pytorch.☆146Updated 6 years ago
- Ordinal regression models in PyTorch☆145Updated 3 years ago
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆160Updated last year
- ECML 2019: Graph Neural Networks for Multi-Label Classification☆90Updated 8 months ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 3 years ago
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆551Updated 3 years ago
- Loss and accuracy go opposite ways...right?☆91Updated 4 years ago
- Simple pytorch implementation of focal loss☆85Updated last year
- Learning Rate Warmup in PyTorch☆404Updated 3 weeks ago
- PyTorch implementation of the GradNorm☆87Updated 6 months ago
- A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quan…☆621Updated 2 years ago
- This repository is for coding exercises listed in Book Hands on One Shot Learning.☆110Updated last year
- The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships" [TMLR 2022].☆133Updated 2 years ago
- Pseudo Labeling for Neural Networks and Logistic Regression/SVMs ( Based on "Pseudo-Label : The Simple and Efficient Semi-Supervised Lear…☆73Updated 5 years ago
- Unofficial PyTorch implementation of Fastformer based on paper "Fastformer: Additive Attention Can Be All You Need"."☆134Updated 3 years ago
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆261Updated 2 years ago
- Code and website for DAL (Discriminative Active Learning) - a new active learning algorithm for neural networks in the batch setting. For…☆202Updated 5 years ago
- This Module implements Spatial Pyramid Pooling (SPP) and Temporal Pyramid Pooling (TPP) as described in different papers.☆52Updated 5 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago