search

SimonVandenhende / Multi-Task-Learning-PyTorch

PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).
Other
752 stars 114 forks source link

Implemention of task balancing methods #16

Closed ifzhang closed 3 years ago

ifzhang commented 3 years ago

Thank you very much for releasing the code! Is there any implemention of the task balancing methods like Uncertainty, GradNorm and DWA? Thanks!

ifzhang commented 3 years ago

@SimonVandenhende