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
Sorting:
- Official implementation of Auxiliary Learning by Implicit Differentiation [ICLR 2021]☆87Jul 25, 2024Updated last year
- 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
- 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 2 years ago
- Adaptive Activation Network and Functional Regularization for Efficient and Flexible Deep Multi-Task Learning☆17Feb 25, 2020Updated 6 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆181Nov 2, 2021Updated 4 years ago
- Knowledge Distillation for Multi-task Learning (ECCV20 Workshops)☆79Apr 7, 2023Updated 2 years ago
- The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].☆729Feb 7, 2022Updated 4 years ago
- Code for "Gradient Surgery for Multi-Task Learning"☆349Apr 28, 2020Updated 5 years ago
- Asymmetric Multi-Task Learning code, If you want to use it, please let me know and cite AMTL paper☆11Aug 3, 2016Updated 9 years ago
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆566Nov 1, 2021Updated 4 years ago
- A deep learning framework for building multimodal multi-task learning systems.☆113Jul 25, 2023Updated 2 years ago
- a much more complex case using GradNorm, where the layer sharing situation is sophisticated.☆15Feb 21, 2019Updated 7 years ago
- A PyTorch implementation of DenseNet, supporting multiclass and multilabel classification.☆24Aug 11, 2017Updated 8 years ago
- AdaTask: A Task-Aware Adaptive Learning Rate Approach to Multi-Task Learning. AAAI, 2023.☆29Sep 29, 2023Updated 2 years ago
- Unofficial implementation of the paper 'Adversarial Training for Free'☆23May 8, 2019Updated 6 years ago
- implementation of our self-guided and self-regularized actor-critic algorithm☆30Jan 1, 2023Updated 3 years ago
- Awesome Multitask Learning Resources☆674Apr 13, 2021Updated 4 years ago
- [CVPR 2020] MTL-NAS: Task-Agnostic Neural Architecture Search towards General-Purpose Multi-Task Learning☆94Dec 8, 2022Updated 3 years ago
- Code for Neural Information Processing Systems (NeurIPS) 2019 paper: Pareto Multi-Task Learning☆143Jan 2, 2020Updated 6 years ago
- Code release for "Understanding Bias in Large-Scale Visual Datasets"☆22Dec 4, 2024Updated last year
- Stochastic AUC Maximization with Deep Neural Networks☆17Apr 1, 2021Updated 4 years ago
- A Keras inspired training utility for PyTorch☆38Sep 13, 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
- [ICML 2020] Efficient Continuous Pareto Exploration in Multi-Task Learning☆149Aug 12, 2021Updated 4 years ago
- ☆67Feb 16, 2023Updated 3 years ago
- Code for "Molecular Hypergraph Grammar with Its Application to Molecular Optimization"☆39Dec 10, 2024Updated last year
- A multi-task learning example for the paper https://arxiv.org/abs/1705.07115☆870Jul 6, 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
- This repository contains the official code for "Flexible Biometrics Recognition: Bridging the Multimodality Gap through Attention, Alignm…☆11Oct 9, 2024Updated last year
- ☆11Mar 13, 2023Updated 2 years ago
- An Efficient Dataset Condensation Plugin and Its Application to Continual Learning. NeurIPS, 2023.☆12Nov 29, 2023Updated 2 years ago
- Code for using the Grasp Affordance Reasoning dataset☆10Sep 17, 2019Updated 6 years ago
- PyTorch implementation of the SIESTA algorithm from our TMLR-2023 paper "SIESTA: Efficient Online Continual Learning with Sleep"☆13Oct 25, 2024Updated last year
- Acoustic event detection using recurrent neural networks.☆11Sep 4, 2018Updated 7 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 list of multi-task machine learning papers.☆37Nov 4, 2018Updated 7 years ago
- ☆31Jan 7, 2023Updated 3 years ago
- A Unified Object Motion and Affinity Model for Online Multi-Object Tracking (CVPR2020)☆93Jun 17, 2024Updated last year