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))