modify Early_Stop

2021-08-09 11:21:18 +08:00
parent c3dc13d2c8
commit 44d1610296
4 changed files with 4 additions and 4 deletions
--- a/fit/tools/train.py
+++ b/fit/tools/train.py
@ -25,7 +25,7 @@ class Early_Stop:
        if delta >= self.eps:
            self.satis_num += 1
        else:
-            self.satis_num = max(0,self.satis_num-1)
+            self.satis_num = 0
        return update_res, self.satis_num >= self.stop_threshold