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
- ☆144Updated 3 years ago
- A list of multi-task learning papers and projects.☆369Updated 3 years ago
- PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).☆790Updated 3 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆787Updated 11 months ago
- Metrics for model evaluation in pytorch☆113Updated 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
- Domain adaptation made easy. Fully featured, modular, and customizable.☆362Updated 2 years ago
- This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow☆136Updated 6 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆172Updated 3 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆256Updated 3 years ago
- ☆51Updated 4 years ago
- A pytorch dataset sampler for always sampling balanced batches.☆111Updated 4 years ago
- Simple pytorch implementation of focal loss☆84Updated last year
- Implementation of Online Label Smoothing in PyTorch☆95Updated 2 years ago
- The net:cal calibration framework is a Python 3 library for measuring and mitigating miscalibration of uncertainty estimates, e.g., by a …☆353Updated 5 months ago
- 🛠 Toolbox to extend PyTorch functionalities☆418Updated 8 months ago
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆259Updated 2 years ago
- CORAL and CORN implementations for ordinal regression with deep neural networks.☆230Updated last year
- The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].☆682Updated 2 years ago
- Learning Rate Warmup in PyTorch☆401Updated 2 weeks ago
- Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning☆595Updated 4 years ago
- Experimenting with different regression losses. Implemented in Pytorch.☆145Updated 6 years ago
- Custom loss functions to use in (mainly) PyTorch.☆38Updated 4 years ago
- Unofficial PyTorch implementation of Fastformer based on paper "Fastformer: Additive Attention Can Be All You Need"."☆134Updated 3 years ago
- PyTorch conversion of https://omoindrot.github.io/triplet-loss☆212Updated last year
- ECML 2019: Graph Neural Networks for Multi-Label Classification☆90Updated 6 months ago
- [NeurIPS 2020] Semi-Supervision (Unlabeled Data) & Self-Supervision Improve Class-Imbalanced / Long-Tailed Learning☆742Updated 3 years ago
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆549Updated 3 years ago
- Flexible components pairing 🤗 Transformers with Pytorch Lightning☆612Updated 2 years ago
- Ranger deep learning optimizer rewrite to use newest components☆326Updated 11 months ago