diff --git a/mindyolo/models/losses/yolov5_loss.py b/mindyolo/models/losses/yolov5_loss.py
index 890cefef..65356ecf 100644
--- a/mindyolo/models/losses/yolov5_loss.py
+++ b/mindyolo/models/losses/yolov5_loss.py
@@ -130,6 +130,8 @@ def construct(self, p, targets, imgs):  # predictions, targets
         bs = p[0].shape[0]  # batch size
 
         loss = lbox + lobj + lcls
+        # Make sure lcls is mindspore tensor
+        lcls = get_tensor(lcls)
         loss_item = ops.stop_gradient(ops.stack((loss, lbox, lobj, lcls)))
         return loss * bs, loss_item