chrisby / torchMTL
A lightweight module for Multi-Task Learning in pytorch.
☆158Updated last year
Related projects ⓘ
Alternatives and complementary repositories for torchMTL
- ☆144Updated 2 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- A list of multi-task learning papers and projects.☆362Updated 2 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆785Updated 9 months ago
- An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.☆225Updated 10 months ago
- Metrics for model evaluation in pytorch☆110Updated 3 years ago
- PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).☆780Updated 2 years 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
- 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
- PyTorch conversion of https://omoindrot.github.io/triplet-loss☆212Updated last year
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆250Updated 2 years ago
- PyTorch implementation of the GradNorm☆72Updated 2 months ago
- This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow☆134Updated 6 years ago
- Unofficial PyTorch implementation of Fastformer based on paper "Fastformer: Additive Attention Can Be All You Need"."☆134Updated 3 years ago
- A pytorch dataset sampler for always sampling balanced batches.☆110Updated 3 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆251Updated 3 years ago
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆547Updated 3 years ago
- Experimenting with different regression losses. Implemented in Pytorch.☆144Updated 5 years ago
- DISTIL: Deep dIverSified inTeractIve Learning. An active/inter-active learning library built on py-torch for reducing labeling costs.☆142Updated last year
- Reproducing the Linear Multihead Attention introduced in Linformer paper (Linformer: Self-Attention with Linear Complexity)☆73Updated 4 years ago
- Learning Rate Warmup in PyTorch☆392Updated this week
- The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].☆677Updated 2 years ago
- A simple and working implementation of Electra, the fastest way to pretrain language models from scratch, in Pytorch☆222Updated last year
- Experiments with supervised contrastive learning methods with different loss functions☆216Updated last year
- ☆51Updated 4 years ago
- Loss function which directly targets ROC-AUC☆241Updated last year
- Implementing PolyLoss in Pytorch☆72Updated 2 years ago
- A basic framework for your PyTorch projects☆73Updated 5 years ago
- Ordinal regression models in PyTorch☆137Updated 2 years ago
- Implementation of Supervised Contrastive Learning with AMP, EMA, SWA, and many other tricks☆166Updated 2 years ago