PyTorch implementation of the GradNorm
-
Updated
Sep 4, 2024 - Jupyter Notebook
#
gradnorm
Here are 4 public repositories matching this topic...
Multi-task learning with adapted torchvision models (SSD, FasterRCNN, DeepLabv3) using Lightning-AI
lightning detection pytorch uncertainty semantic-segmentation fasterrcnn dwa deeplabv3 multitask-learning ssdlite torchvision gradnorm autolambda loss-balancing
-
Updated
May 5, 2023 - Python
Multi-Task GNN for Truck Speed Prediction under Extreme Weather (ACM SIGSPATIAL 2022) — fork of giobbu/AST-MTL with weather auxiliary head + GradNorm
-
Updated
May 14, 2026 - Jupyter Notebook
Improve this page
Add a description, image, and links to the gradnorm topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the gradnorm topic, visit your repo's landing page and select "manage topics."