awl的parameter不更新

Mikoto10032 / AutomaticWeightedLoss

Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning

Apache License 2.0

575 stars 81 forks source link

Open fanbooo opened 2 years ago

fanbooo commented 2 years ago

作者你好：我有2个任务，都用的交叉熵，训练时候发现awl的参数不更新，请问你碰到过这个问题吗？

shuaizzZ commented 2 years ago

我尝试了这个也不更新，貌似需要把model和loss放在一个module里才行

wenshengyoung commented 2 years ago

你解决这个问题了吗我也试的是不更新

ZZX-11 commented 9 months ago

你要把param，放到optimizer里