src/transformers/models/upernet/modeling_upernet.py/UperNetHead/_init_weights
class UperNetHead:    def _init_weights(self, module):
        if isinstance(module, nn.Conv2d):
            module.weight.data.normal_(mean=0.0, std=self.config.initializer_range)
            if module.bias is not None:
                module.bias.data.zero_()