[Fix]: code weights for anchor3d_head (#173)

cec2d8b0 · Liang Liu · GitHub · c84285da · cec2d8b0
Unverified Commit cec2d8b0 authored Oct 20, 2020 by Liang Liu Committed by GitHub Oct 20, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet3d/models/dense_heads/anchor3d_head.py mmdet3d/models/dense_heads/anchor3d_head.py +1 -1

No files found.
--- a/mmdet3d/models/dense_heads/anchor3d_head.py
+++ b/mmdet3d/models/dense_heads/anchor3d_head.py
@@ -244,7 +244,7 @@ class Anchor3DHead(nn.Module, AnchorTrainMixin):
        if num_pos > 0:
            code_weight = self.train_cfg.get('code_weight', None)
            if code_weight:
-                bbox_weights = bbox_weights * bbox_weights.new_tensor(
+                pos_bbox_weights = pos_bbox_weights * bbox_weights.new_tensor(
                    code_weight)
            if self.diff_rad_by_sin:
                pos_bbox_pred, pos_bbox_targets = self.add_sin_difference(