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