search

yaringal / multi-task-learning-example

A multi-task learning example for the paper https://arxiv.org/abs/1705.07115
MIT License
838 stars 205 forks source link

Calculating back to actual weights of loss functions #15

Closed edshkim98 closed 3 years ago

edshkim98 commented 3 years ago

@yaringal Hi, thank you very much for releasing the code here! I am currently using your technique in adversarial loss training for semantic segmentation. However, I would like to know the actual weights applying to each loss function (Cross Entropy, Adversarial Loss). Could you please help me with how to get these weights?

(I am currently using your equation of std and do 1/(2*std**2) to get the weight. Is this correct?)