How to assign different weights to auxiliary task's loss in the whole loss function - multitask

shenweichen / DeepCTR-Torch

【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.

https://deepctr-torch.readthedocs.io/en/latest/index.html

Apache License 2.0

3k stars 703 forks source link

How to assign different weights to auxiliary task's loss in the whole loss function - multitask #270

Open CFZhai opened 1 year ago

CFZhai commented 1 year ago

I cannot find a way to assign different weights to our main task's loss and auxiliary task's loss in the whole loss function. How can we adjust the weights by ourselves?