Yes, just remove the sigmoid and  pass those outputs directly to BCEWithLogitsLoss. Well, u don’t really have to call sigmoid again in Validation because for sigmoid, when x=0, y=0.5, u can use 0 as an indicator.