AutomaticWeightedLoss awl的parameter不更新

awl的parameter不更新

Open fanbooo opened this issue 3 years ago • 3 comments

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

Feb 14 '22 13:02 fanbooo

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

Apr 18 '22 02:04 shuaizzZ

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

Jul 19 '22 14:07 wenshengyoung

你要把param，放到optimizer里

Jan 13 '24 12:01 ZZX-11