copperwiring:

criterion = nn.BCEWithLogitsLoss() #has sigmoid internally


Hi @albanD, I used criterion = nn.BCEWithLogitsLoss() #has sigmoid internally for loss calculation. Pardon me but I didn’t get what exactly did you mean by loss value evaluation