diff --git a/ppcls/loss/multilabelloss.py b/ppcls/loss/multilabelloss.py
index 2dd61d1c7..51ae97838 100644
--- a/ppcls/loss/multilabelloss.py
+++ b/ppcls/loss/multilabelloss.py
@@ -39,7 +39,7 @@ class MultiLabelLoss(nn.Layer):
     def _binary_crossentropy(self, input, target, class_num):
         if self.weight_ratio:
             target, label_ratio = target[:, 0, :], target[:, 1, :]
-        else:
+        elif target.ndim == 3:
             target = target[:, 0, :]
         if self.epsilon is not None:
             target = self._labelsmoothing(target, class_num)