Method __init__

models/dat.py:26–32

(self,
                 dim: int,
                 inplace: bool = False,
                 init_values: float = 1e-5)

class LayerScale(nn.Module):

    def __init__(self,
                 dim: int,
                 inplace: bool = False,
                 init_values: float = 1e-5):
        super().__init__()
        self.inplace = inplace
        self.weight = nn.Parameter(torch.ones(dim) * init_values)

    def forward(self, x):
        if self.inplace: