def METHOD_NAME(self):
    super().METHOD_NAME()
    if not isinstance(self.weight, nn.UninitializedParameter):
        size = self.weight.size(0) * self.weight.size(1)
        uniform(size, self.weight)
    if self.root_weight:
        self.lin.METHOD_NAME()
    zeros(self.bias)