From 44d16102967e66d654d4efe2a63d982ce33e018a Mon Sep 17 00:00:00 2001
From: Iridoudou <2534936416@qq.com>
Date: Mon, 9 Aug 2021 11:21:18 +0800
Subject: [PATCH] modify Early_Stop

---
 fit/configs/HumanAct12.json | 2 +-
 fit/configs/UTD_MHAD.json   | 2 +-
 fit/tools/main.py           | 2 +-
 fit/tools/train.py          | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/fit/configs/HumanAct12.json b/fit/configs/HumanAct12.json
index cbe1207..2e3774b 100644
--- a/fit/configs/HumanAct12.json
+++ b/fit/configs/HumanAct12.json
@@ -4,7 +4,7 @@
     },
     "TRAIN": {
         "LEARNING_RATE": 2e-2,
-        "MAX_EPOCH": 500,
+        "MAX_EPOCH": 1000,
         "WRITE": 1
     },
     "USE_GPU": 1,
diff --git a/fit/configs/UTD_MHAD.json b/fit/configs/UTD_MHAD.json
index fc91d77..a6f51c1 100644
--- a/fit/configs/UTD_MHAD.json
+++ b/fit/configs/UTD_MHAD.json
@@ -4,7 +4,7 @@
     },
     "TRAIN": {
         "LEARNING_RATE": 2e-2,
-        "MAX_EPOCH": 500,
+        "MAX_EPOCH": 1000,
         "WRITE": 1
     },
     "USE_GPU": 1,
diff --git a/fit/tools/main.py b/fit/tools/main.py
index c8527e3..7a12723 100644
--- a/fit/tools/main.py
+++ b/fit/tools/main.py
@@ -12,7 +12,7 @@ sys.path.append(os.getcwd())
 from smplpytorch.pytorch.smpl_layer import SMPL_Layer
 from train import train
 from transform import transform
-from save import save_pic,save_params
+from save import save_params
 from load import load
 torch.backends.cudnn.benchmark=True
 
diff --git a/fit/tools/train.py b/fit/tools/train.py
index 4990177..b8d0ff4 100644
--- 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