early-stopping-pytorch

early-stopping-pytorch copied to clipboard

I've been using this EarlyStopping class for my master's thesis project, but I had to modify it slightly to save the model with the lowest val_loss for a given fold...

Jocareher

Do you think the early stopping condition check in line number `36` in `__cal__()` function of `EarlyStopping` class should be less than or equal to instead of just less than?...

DhruvAwasthi

as title said

Suasy