神经网络搭建(Pytorch)——联合损失函数

今天看论文看到了联合损失函数的概念，查了一下，pytorch下可以通过重写损失函数来达到联合损失函数的效果。

自定义损失函数：

1. 继承nn.Module类
2. 重写forward()方法

class MyLoss(nn.Module):
    def forward(self,output,target):
    	loss1 = ...
        loss2 = ...
		loss = (loss1 + loss2) / 2    # 计算平均值
        return loss