[Fix]: fix HTC Runtime error in PyTorch 1.10 (open-mmlab#8083)

ZwwWayne · Jul 19, 2022 · 2243fd6 · 2243fd6
1 parent 36470ac
commit 2243fd6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmdet/models/roi_heads/mask_heads/fused_semantic_head.py b/mmdet/models/roi_heads/mask_heads/fused_semantic_head.py
@@ -101,7 +101,8 @@ def forward(self, feats):
             if i != self.fusion_level:
                 feat = F.interpolate(
                     feat, size=fused_size, mode='bilinear', align_corners=True)
-                x += self.lateral_convs[i](feat)
+                # fix runtime error of "+=" inplace operation in PyTorch 1.10
+                x = x + self.lateral_convs[i](feat)
 
         for i in range(self.num_convs):
             x = self.convs[i](x)