chrisby / torchMTL
A lightweight module for Multi-Task Learning in pytorch.
☆163Updated last year
Alternatives and similar repositories for torchMTL:
Users that are interested in torchMTL are comparing it to the libraries listed below
- Metrics for model evaluation in pytorch☆113Updated 4 years ago
- ☆146Updated 3 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- A list of multi-task learning papers and projects.☆370Updated 3 years ago
- Simple pytorch implementation of focal loss☆85Updated 2 years ago
- Ordinal regression models in PyTorch☆147Updated 3 years ago
- CORAL and CORN implementations for ordinal regression with deep neural networks.☆232Updated last year
- Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.☆149Updated 5 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 3 years ago
- PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).☆804Updated 3 years ago
- An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.☆235Updated last year
- Experimenting with different regression losses. Implemented in Pytorch.☆147Updated 6 years ago
- Provides everything needed for high performance data loading and augmentation in pytorch.☆319Updated last year
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆160Updated last year
- Learning Rate Warmup in PyTorch☆409Updated last month
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆552Updated 3 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆796Updated last year
- A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quan…☆624Updated 2 years ago
- A pytorch dataset sampler for always sampling balanced batches.☆114Updated 4 years ago
- Ranger deep learning optimizer rewrite to use newest components☆329Updated last year
- Custom loss functions to use in (mainly) PyTorch.☆39Updated 4 years ago
- Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.☆235Updated 3 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆176Updated 3 years ago
- DISTIL: Deep dIverSified inTeractIve Learning. An active/inter-active learning library built on py-torch for reducing labeling costs.☆149Updated 2 years ago
- Domain adaptation made easy. Fully featured, modular, and customizable.☆373Updated 2 years ago
- SpinalNet: Deep Neural Network with Gradual Input☆173Updated 10 months ago
- The net:cal calibration framework is a Python 3 library for measuring and mitigating miscalibration of uncertainty estimates, e.g., by a …☆357Updated 8 months ago
- PyTorch implementation of the GradNorm☆92Updated 7 months ago
- Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for interdisciplinary research, part of the 🔥…☆457Updated this week
- Pseudo Labeling for Neural Networks and Logistic Regression/SVMs ( Based on "Pseudo-Label : The Simple and Efficient Semi-Supervised Lear…☆73Updated 5 years ago