valid_train (#3567)

PaddlePaddle · Nov 8, 2023 · e167b03 · e167b03
1 parent 12ca7e3
commit e167b03
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/paddleseg/core/train.py b/paddleseg/core/train.py
@@ -65,7 +65,7 @@ def train(model,
           save_dir='output',
           iters=10000,
           batch_size=2,
-          early_stop_interval=None,
+          early_stop_intervals=None,
           resume_model=None,
           save_interval=1000,
           log_iters=10,
@@ -367,7 +367,7 @@ def train(model,
                     elif mean_iou < best_mean_iou:
                         stop_count += 1
 
-                    if early_stop_interval is not None and stop_count >= early_stop_interval:
+                    if early_stop_intervals is not None and stop_count >= early_stop_intervals:
                         stop_status = True
                         logger.info(
                             'Early stopping at iter {}. The best mean IoU is {:.4f}.'

diff --git a/tools/train.py b/tools/train.py
@@ -192,7 +192,7 @@ def main(args):
         save_dir=args.save_dir,
         iters=cfg.iters,
         batch_size=cfg.batch_size,
-        early_stop=args.early_stop_intervals,
+        early_stop_intervals=args.early_stop_intervals,
         resume_model=args.resume_model,
         save_interval=args.save_interval,
         log_iters=args.log_iters,