def forward(self, x):
    x = x.flatten(start_dim=1)
    x = F.relu(self.fc6(x))
    x = F.relu(self.fc7(x))
    return x
def forward(self, x):
    x = x.flatten(start_dim=1)
    x = F.relu(self.fc6(x))
    x = F.relu(self.fc7(x))
    return x