Method init_weights

mmpose/models/necks/fpn.py:148–152

Initialize model weights.

(self)

                self.fpn_convs.append(extra_fpn_conv)

    def init_weights(self):
        """Initialize model weights."""
        for m in self.modules():
            if isinstance(m, nn.Conv2d):
                xavier_init(m, distribution='uniform')

    def forward(self, inputs):
        """Forward function."""