diff --git a/models/deepcrack_model.py b/models/deepcrack_model.py
index 00ed709a..0956cb86 100644
--- a/models/deepcrack_model.py
+++ b/models/deepcrack_model.py
@@ -98,6 +98,7 @@ def backward(self):
         lambda_fused = self.opt.lambda_fused
 
         self.loss_side = 0.0
+        self.label = self.label.float() 
         for out, w in zip(self.outputs[:-1], self.weight_side):
             #self.loss_side += self.criterionSeg(out, self.label3d) * w
             self.loss_side += self.criterionSeg(out, self.label) * w