From 9979b268af7317034abb7c880afb83b05dc9d772 Mon Sep 17 00:00:00 2001 From: vincentme <11166542+vincentme@users.noreply.github.com> Date: Fri, 21 May 2021 10:43:58 -0500 Subject: [PATCH] Update combined.py fix the bug of compute combined loss --- pymic/loss/seg/combined.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pymic/loss/seg/combined.py b/pymic/loss/seg/combined.py index 9102d37..1a75874 100644 --- a/pymic/loss/seg/combined.py +++ b/pymic/loss/seg/combined.py @@ -22,5 +22,5 @@ def __init__(self, params, loss_dict): def forward(self, loss_input_dict): loss_value = 0.0 for i in range(len(self.loss_list)): - loss_value = self.loss_weight[i] + self.loss_list[i](loss_input_dict) - return loss_value \ No newline at end of file + loss_value += self.loss_weight[i]*self.loss_list[i](loss_input_dict) + return loss_value