human
2. Fully Connected Model
 ... Write me a the Python program you requested?
assistant
#Here is the Python program you requested.
data = get_data(32, bs)
lr = 1e-2

class SimpleNet(nn.Module):
    def __init__(self, layers):
        super().__init__()
        self.layers = nn.ModuleList([
            nn.Linear(layers[i], layers[i+1]) for i in range(len(layers) -1 )])
    def forward(self, x):
        x = x.view(x.size(0), -1) # input tensor shape
        for λ in self.layers:     # for each layer
            λ_x = λ(x)            # its output is itself acted upon the input
            x = F.relu(λ_x) # "nonlinearity" (activation function)
        return F.log_softmax(λ_x, dim=1) # output activation function