From a93dfbf14ff24f26fa47d5c85a4b0e227ca95719 Mon Sep 17 00:00:00 2001 From: ManoleAlexandru99 Date: Thu, 23 Mar 2023 20:23:53 +0200 Subject: [PATCH] Enabled multitask with weights 1, 1 #0005 --- utils/loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loss.py b/utils/loss.py index 14eed55c665f..364327e19186 100644 --- a/utils/loss.py +++ b/utils/loss.py @@ -185,7 +185,7 @@ def __call__(self, preds, targets, seg_masks): # predictions, targets # return (lbox + lobj + lcls) * bs, torch.cat((lbox, lobj, lcls)).detach() # return total_loss, torch.cat((lbox, lobj, lcls, lseg)).detach() - return (lbox + lobj + lcls) * bs * 0, lseg * bs * 1, torch.cat((lbox, lobj, lcls, lseg)).detach() + return (lbox + lobj + lcls) * bs * 1, lseg * bs * 1, torch.cat((lbox, lobj, lcls, lseg)).detach() def build_targets(self, p, targets): # Build targets for compute_loss(), input targets(image,class,x,y,w,h)