From 78a90c9661a05e8e1b7cc52a0989e4895fd96981 Mon Sep 17 00:00:00 2001 From: Sheng Hu Date: Thu, 23 Mar 2023 18:29:44 +0800 Subject: [PATCH] Remove duplicate assignment code (#11178) Signed-off-by: Sheng Hu --- utils/segment/loss.py | 1 - 1 file changed, 1 deletion(-) diff --git a/utils/segment/loss.py b/utils/segment/loss.py index 2a8a4c6..caeff3c 100644 --- a/utils/segment/loss.py +++ b/utils/segment/loss.py @@ -16,7 +16,6 @@ class ComputeLoss: self.overlap = overlap device = next(model.parameters()).device # get model device h = model.hyp # hyperparameters - self.device = device # Define criteria BCEcls = nn.BCEWithLogitsLoss(pos_weight=torch.tensor([h['cls_pw']], device=device))