Commit 8248443a authored by Andrei Roibu's avatar Andrei Roibu
Browse files

updated patience to 5 from 2

parent de062ac3
......@@ -139,9 +139,9 @@ class Solver():
if use_last_checkpoint:
self.load_checkpoint()
self.EarlyStopping = EarlyStopping(patience=2, min_delta=0, best_score=self.best_score_early_stop, counter=self.counter_early_stop)
self.EarlyStopping = EarlyStopping(patience=5, min_delta=0, best_score=self.best_score_early_stop, counter=self.counter_early_stop)
else:
self.EarlyStopping = EarlyStopping(patience=2, min_delta=0)
self.EarlyStopping = EarlyStopping(patience=5, min_delta=0)
def train(self, train_loader, validation_loader):
"""Training Function
......
Markdown is supported
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