Fix the bug: 'Upsample' object has no attribute 'weight' (open-mmlab#…

…6378)
jozhang97 · Oct 27, 2021 · 2cd24a2 · 2cd24a2
1 parent 4c563a0
commit 2cd24a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/models/roi_heads/mask_heads/fcn_mask_head.py b/mmdet/models/roi_heads/mask_heads/fcn_mask_head.py
@@ -119,7 +119,7 @@ def init_weights(self):
                 continue
             elif isinstance(m, CARAFEPack):
                 m.init_weights()
-            else:
+            elif hasattr(m, 'weight') and hasattr(m, 'bias'):
                 nn.init.kaiming_normal_(
                     m.weight, mode='fan_out', nonlinearity='relu')
                 nn.init.constant_(m.bias, 0)