Method forward

train.py:103–104

(self, x)

            nn.Linear(h2, 1)
        )
    def forward(self, x):
        return self.net(x)

model = MLP().to(device)
criterion = nn.BCEWithLogitsLoss()