Toy example in pytorch for binary classification

Hello PyTorch.ipynb

Hello Santi, thank you for this code snippet! It really helped me start understanding how to implement pytorch for classification.

I wanted to point out an error in your code regarding the train_epoch function:

1. for the Forward function call, you write:
   y_hat = net(x_batch)
   Where 'net' should actually be 'model' (since this was the argument passed into train_epoch function).
2. It would be better if you actually had the argument X,Y defined as arguments in the train_epoch function rather than calling the global variables X and Y.

Anyway, thanks for putting this up!

BCEWithLogitsLoss() expects logits as inputs and not probabilities, so you shouldnt do y = self.out_act(a3)

BCEWithLogitsLoss() expects logits as inputs and not probabilities, so you shouldnt do y = self.out_act(a3)

But he's using BCELoss not BCEWithLogitsLoss