Method __init__

models/linear.py:98–103

(self, n_class=10, bayesian=False)

# net
class LeNet(nn.Module):
    def __init__(self, n_class=10, bayesian=False):
        super(LeNet, self).__init__()
        self.feature_extractor = Net1_fea()
        self.linear = Net1_clf(n_class)
        self.discriminator = Net1_dis()
        self.bayesian = bayesian
    
    def forward(self, x, intermediate=False):
        x, in_values = self.feature_extractor(x)