I used 5 epochs to train the model whereas they used 300, does it matter? Even with just 5 I should still be seeing some results.
After debugging the code I see that the dataloader lists my malicious class (my two classes are “malicious” and “benign”) of my training dataloader as follows:

image.png1005×66 3.94 KB

Is the timeout significant? I never noticed it before, and have no idea why its just for that class.
But everything still seems to work, The very first training batch of 8 samples of both classes gives this as output (after evaluated with torch.exp(y_model)):

Does this mean the samples do have some effect on the training? Still not sure what is going, however it looks like the model is getting all the data fed in correctly.
I tried to overfit the model on a small sample size but it still only outputs 50% accuracy. I have no idea what is going on.