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