PyTorch implementation of the GradNorm
☆127Sep 4, 2024Updated last year
Alternatives and similar repositories for GradNorm
Users that are interested in GradNorm are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆272Aug 9, 2022Updated 3 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
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆133Jun 1, 2026Updated last month
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆403Jun 22, 2021Updated 5 years ago
- Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning☆649Jun 20, 2020Updated 6 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- A PyTorch Library for Multi-Task Learning☆2,573May 14, 2025Updated last year
- a much more complex case using GradNorm, where the layer sharing situation is sophisticated.☆16Feb 21, 2019Updated 7 years ago
- Objective metrics for measuring visual texture similarity using STSIM features. Supervised by Thrasos Pappas.☆14Oct 4, 2023Updated 2 years ago
- Enhancing the convergence speed by 2x and improving the training success of Physics-Informed Neural Networks (PINNs).☆13Oct 14, 2024Updated last year
- The repository contains the MATLAB-based model of IEEE 802.11ad physical layer in baseband.☆22Jan 27, 2020Updated 6 years ago
- A PyTorch implementation of BatchBALD on the MNIST dataset☆13Sep 16, 2020Updated 5 years ago
- Deep Attentive Time Warping in Pattern Recognition 2023.☆14Dec 10, 2023Updated 2 years ago
- ☆11Nov 14, 2020Updated 5 years ago
- Accompanyig code for "Training Physics-Informed Neural Networks: one learning to rule them all?"☆13Nov 15, 2022Updated 3 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆15Dec 13, 2022Updated 3 years ago
- A Large-Scale Dataset and Hierarchical Neural Solver for Data-Driven Structural Crash Simulation☆17May 14, 2026Updated 2 months ago
- The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships" [TMLR 2022].☆140Feb 5, 2023Updated 3 years ago
- Deep learning techniques for atherosclerotic plaque detection in the carotid artery☆16Jun 16, 2022Updated 4 years ago
- A list of papers, codes and applications on multi-task learning.☆80Oct 1, 2025Updated 9 months ago
- An up-to-date list of works on Multi-Task Learning☆378Mar 2, 2026Updated 4 months ago
- [ECCV2024] Domesticating SAM for Breast Ultrasound Image Segmentation via Spatial-frequency Fusion and Uncertainty Correction☆18Jul 12, 2024Updated 2 years ago
- A simple method of vehicle counting and speed prediction☆13Jun 14, 2020Updated 6 years ago
- Dataset and benchmark for indoor monocular depth estimation with focus on indoor space types☆17Dec 7, 2024Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Code accompanying the paper "TiZero: Mastering Multi-Agent Football with Curriculum Learning and Self-Play" (AAMAS 2023) 足球游戏智能体☆14May 25, 2023Updated 3 years ago
- MTAdam: Automatic Balancing of Multiple Training Loss Terms☆37Nov 21, 2020Updated 5 years ago
- A pytorch version of hamiltonian monte carlo☆15Jun 26, 2019Updated 7 years ago
- This repo contains variations of Orthogonal Matching Pursuit (OMP) Algorithm family. The family includes: Piecewise Orthogonal Matching P…☆22Jun 9, 2021Updated 5 years ago
- All code and data necessary to replicate experiments in the paper BAGM: A Backdoor Attack for Manipulating Text-to-Image Generative Model…☆13Sep 16, 2024Updated last year
- Early sepsis detection w/ multitask gaussian process neural network☆15Sep 20, 2019Updated 6 years ago
- PyTorch implementation of popular attention mechanisms in vision☆20Sep 8, 2025Updated 10 months ago
- Short Term Load Forecasting with ES-adRNN☆17Dec 8, 2022Updated 3 years ago
- Controlling a mobile robot using a Logitech F710 gamepad☆15Jun 24, 2026Updated 3 weeks 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.
- Coordinate Descent Fuzzy Twin Support Vector Machine for Classification☆11Jan 13, 2018Updated 8 years ago
- Deep Learning for the Detection and Characterization of the Carotid Artery in Ultrasound Imaging☆18Sep 24, 2022Updated 3 years ago
- A curated list of DATASETS, CODEBASES and PAPERS on Multi-Task Learning (MTL), from Machine Learning perspective.☆842Mar 3, 2026Updated 4 months ago
- ☆14Jul 27, 2021Updated 4 years ago
- The MATLAB code below implements the second-order SPSA (simultaneous perturbation stochastic approximation) and second-order SG (stochast…☆13Nov 27, 2020Updated 5 years ago
- mamba2-jax: A pure JAX/Flax implementation of Mamba-2 for language modeling and time series forecasting.☆16Jun 23, 2026Updated 3 weeks ago
- DyEdgeGAT: Dynamic Edge via Graph Attention for Early Fault Detection in IIoT Systems☆20Mar 7, 2026Updated 4 months ago