Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.
☆150Jun 3, 2019Updated 6 years ago
Alternatives and similar repositories for Hydra
Users that are interested in Hydra are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Source code for Neural Information Processing Systems (NeurIPS) 2018 paper "Multi-Task Learning as Multi-Objective Optimization"☆1,070Sep 2, 2024Updated last year
- Early sepsis detection w/ multitask gaussian process neural network☆15Sep 20, 2019Updated 6 years ago
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆273Aug 9, 2022Updated 3 years ago
- ☆148Dec 13, 2021Updated 4 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆182Nov 2, 2021Updated 4 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Knowledge Distillation for Multi-task Learning (ECCV20 Workshops)☆78Apr 7, 2023Updated 3 years ago
- The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].☆730Feb 7, 2022Updated 4 years ago
- Code for "Gradient Surgery for Multi-Task Learning"☆356Apr 28, 2020Updated 5 years ago
- This project demonstrates the use of generic bi-directional LSTM models for predicting importance of words in a spoken dialgoue for under…☆10Mar 24, 2023Updated 3 years ago
- [CVPR 2020] MTL-NAS: Task-Agnostic Neural Architecture Search towards General-Purpose Multi-Task Learning☆94Dec 8, 2022Updated 3 years ago
- Official Implementation of "Doubly Mixed-Effects Gaussian Process Regression" (Jun Ho Yoon, Daniel P. Jeong, Seyoung Kim) (AISTATS 2022, …☆12Jul 13, 2022Updated 3 years ago
- A deep learning framework for building multimodal multi-task learning systems.☆113Jul 25, 2023Updated 2 years ago
- Code for Neural Information Processing Systems (NeurIPS) 2019 paper: Pareto Multi-Task Learning☆143Jan 2, 2020Updated 6 years ago
- A multi-task learning example for the paper https://arxiv.org/abs/1705.07115☆870Jul 6, 2020Updated 5 years ago
- Wordpress hosting with auto-scaling - Free Trial • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- [ICML 2020] Efficient Continuous Pareto Exploration in Multi-Task Learning☆149Aug 12, 2021Updated 4 years ago
- a much more complex case using GradNorm, where the layer sharing situation is sophisticated.☆16Feb 21, 2019Updated 7 years ago
- Awesome Multitask Learning Resources☆672Apr 13, 2021Updated 5 years ago
- PyTorch implementation for "Gradient Surgery for Multi-Task Learning" https://arxiv.org/abs/2001.06782☆14Jul 6, 2020Updated 5 years ago
- A PyTorch implementation of DenseNet, supporting multiclass and multilabel classification.☆24Aug 11, 2017Updated 8 years ago
- A Tensorflow implementation of the paper arXiv:1604.03539☆133Jul 25, 2018Updated 7 years ago
- Jarvis is a toolbox built on top of TensorFlow2.0 that allows developers and researchers to easily build neural networks in TensorFlow, p…☆37Oct 15, 2020Updated 5 years ago
- This is the official implementation for COSMOS: a method to learn Pareto fronts that scales to large datasets and deep models.☆39May 25, 2021Updated 4 years ago
- Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning☆648Jun 20, 2020Updated 5 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- ☆67Feb 16, 2023Updated 3 years ago
- A TensorFlow Keras implementation of "Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts" (KDD 2018)☆736Mar 25, 2023Updated 3 years ago
- Tensorflow implementation of Deep Asymmetric Multi-task Feature Learning (ICML2018)☆14Sep 12, 2018Updated 7 years ago
- A lightweight module for Multi-Task Learning in pytorch.☆168Aug 7, 2023Updated 2 years ago
- Official Implementation of TILDE-Q: A Transformation Invariant Loss Function for Time-Series Forecasting☆22Feb 11, 2025Updated last year
- Stochastic AUC Maximization with Deep Neural Networks☆17Apr 1, 2021Updated 5 years ago
- TensorFlow code for the paper Convolutional RNN: an Enhanced Model for Extracting Features from Sequential Data (https://arxiv.org/abs/16…☆32May 29, 2017Updated 8 years ago
- An implementation of RaFM. Xiaoshuang Chen, Yin Zheng, Jiaxiang Wang, et al. "RaFM: Rank-Aware Factorization Machines"☆12May 11, 2019Updated 6 years ago
- A Keras inspired training utility for PyTorch☆38Sep 13, 2018Updated 7 years ago
- Deploy open-source AI quickly and easily - Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Official implementation of "Multi-Task Learning as a Bargaining Game" [ICML 2022]☆239Jun 25, 2025Updated 9 months ago
- The implementation of "Self-Supervised Generalisation with Meta Auxiliary Learning" [NeurIPS 2019].☆175Dec 19, 2021Updated 4 years ago
- ☆11Oct 9, 2021Updated 4 years ago
- Files for the paper: "Sound Source Localization using Deep Residual Learning"☆24Nov 13, 2017Updated 8 years ago
- [Code] Deep Multi-task Representation Learning: A Tensor Factorisation Approach☆57Jul 4, 2017Updated 8 years ago
- A list of multi-task learning papers and projects.☆385Jan 31, 2022Updated 4 years ago
- PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).☆840Jan 13, 2022Updated 4 years ago