Method __init__

models/linear.py:9–12

(self, dim, n_class)

class linMod(nn.Module):
    def __init__(self, dim, n_class):
        super(linMod, self).__init__()
        self.dim = dim
        self.lm = nn.Linear(int(np.prod(dim)), n_class)
    def forward(self, x):
        x = x.view(-1, int(np.prod(self.dim)))
        out = self.lm(x)