fix broadcast shape bug in yolov3 (open-mmlab#7551)

ZwwWayne · Jul 19, 2022 · 3d77502 · 3d77502
1 parent ccce2d9
commit 3d77502
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mmdet/models/losses/cross_entropy_loss.py b/mmdet/models/losses/cross_entropy_loss.py
@@ -123,7 +123,10 @@ def binary_cross_entropy(pred,
         # should mask out the ignored elements
         valid_mask = ((label >= 0) & (label != ignore_index)).float()
         if weight is not None:
-            weight *= valid_mask
+            # The inplace writing method will have a mismatched broadcast
+            # shape error if the weight and valid_mask dimensions
+            # are inconsistent such as (B,N,1) and (B,N,C).
+            weight = weight * valid_mask
         else:
             weight = valid_mask