INNER CODE UNIT · Python

_init_weight

pprp/awesome-attention-mechanism-in-cv · Plug-and-Play-Module/ASPPBlock.py:74

    def _init_weight(self):
        for m in self.modules():
            if isinstance(m, nn.Conv2d):
                torch.nn.init.kaiming_normal_(m.weight)
            elif isinstance(m, nn.BatchNorm2d):
                m.weight.data.fill_(1)
                m.bias.data.zero_()

View source record →