SimonVandenhende / Multi-Task-Learning-PyTorch

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

About task balancing #8

Open lelan-li opened 3 years ago

lelan-li commented 3 years ago

Thank you for your excellent work and open-source code.

Could you please provide some code about task balancing, such as uncertainty weight, GradNorm multi-object Optim? Your paper has some experimental results about these methods. Thank you very much!

SimonVandenhende commented 3 years ago

Currently, I am not planning to release the code for the task balancing strategies, as I am involved in other projects. Still, I am available to answer any questions you might have. Also, note that the fixed weighting strategy gives you a strong baseline, which future research should try to improve.

Li-Qingyun commented 2 years ago

I'm also looking forward to the open-source code of the optimizer strategies.