小编Kon*_*kos的帖子

如何在pytorch中处理多重损失?

在此输入图像描述

比如这个,我想用一些辅助损失来提升我的模特表现.
哪个类型代码可以在pytorch中实现它?

#one
loss1.backward()
loss2.backward()
loss3.backward()
optimizer.step()
#two
loss1.backward()
optimizer.step() 
loss2.backward()
optimizer.step() 
loss3.backward()
optimizer.step()   
#three
loss = loss1+loss2+loss3
loss.backward()
optimizer.step()
Run Code Online (Sandbox Code Playgroud)

感谢您的回答!

python pytorch

9
推荐指数
3
解决办法
3261
查看次数

标签 统计

python ×1

pytorch ×1