def forward(self, x):
    x = F.relu(self.lin1(x))
    x = self.dropout(x)
    x = self.lin2(x)
    x = self.dropout(x)
    return x
def forward(self, x):
    x = F.relu(self.lin1(x))
    x = self.dropout(x)
    x = self.lin2(x)
    x = self.dropout(x)
    return x