def METHOD_NAME(self):
    layers = [
        *self.additional_blocks,
        *self.loc, *self.conf]
    for layer in layers:
        for param in layer.parameters():
            if param.dim() > 1: nn.init.xavier_uniform_(param)