Commit d12bcf8e authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

fixed bug preventing stop if score=best_score

parent 52785b10
......@@ -50,7 +50,7 @@ class EarlyStopping:
self.best_score = score
self.save_checkpoint = True
elif score < self.best_score:
elif score <= self.best_score:
self.counter += 1
self.save_checkpoint = False
print("Early Stopping Counter: {}/{}".format(self.counter, self.patience))
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment