[TIL]Day 56

MLP 성능 향상

히든레이어 추가

    self.fc1 = nn.Linear(28 * 28, 512)
    self.fc2 = nn.Linear(512,14*14)
    self.fc3 = nn.Linear(14*14, 7*7)
    self.fc4 = nn.Linear(7*7,10)
    
    x = F.relu(self.fc1(x))
    x = F.relu(self.fc2(x))
    x = F.relu(self.fc3(x))