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(): ...@@ -139,9 +139,9 @@ class Solver():
if use_last_checkpoint: if use_last_checkpoint:
self.load_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: else:
self.EarlyStopping = EarlyStopping(patience=2, min_delta=0) self.EarlyStopping = EarlyStopping(patience=5, min_delta=0)
def train(self, train_loader, validation_loader): def train(self, train_loader, validation_loader):
"""Training Function """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