From df6ccb19160fd36af19313bffe6b44118fdede59 Mon Sep 17 00:00:00 2001
From: michaelowenliu <liuyi.ntu@gmail.com>
Date: Wed, 23 Sep 2020 17:17:06 +0800
Subject: [PATCH] change eval_dataset to val_dataset

---
 dygraph/paddleseg/core/train.py | 8 ++++----
 dygraph/train.py                | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/dygraph/paddleseg/core/train.py b/dygraph/paddleseg/core/train.py
index d1854a0b..476dfce3 100644
--- a/dygraph/paddleseg/core/train.py
+++ b/dygraph/paddleseg/core/train.py
@@ -52,7 +52,7 @@ def loss_computation(logits, label, losses):
 def train(model,
           train_dataset,
           places=None,
-          eval_dataset=None,
+          val_dataset=None,
           optimizer=None,
           save_dir='output',
           iters=10000,
@@ -173,10 +173,10 @@ def train(model,
                 paddle.save(optimizer.state_dict(),
                             os.path.join(current_save_dir, 'model'))
 
-                if eval_dataset is not None:
+                if val_dataset is not None:
                     mean_iou, avg_acc = evaluate(
                         model,
-                        eval_dataset,
+                        val_dataset,
                         model_dir=current_save_dir,
                         num_classes=num_classes,
                         ignore_index=ignore_index,
@@ -188,7 +188,7 @@ def train(model,
                         paddle.save(model.state_dict(),
                                     os.path.join(best_model_dir, 'model'))
                     logger.info(
-                        'Current evaluated best model in eval_dataset is iter_{}, miou={:4f}'
+                        'Current evaluated best model in val_dataset is iter_{}, miou={:4f}'
                         .format(best_model_iter, best_mean_iou))
 
                     if use_vdl:
diff --git a/dygraph/train.py b/dygraph/train.py
index ba60a5aa..5a0b66c3 100644
--- a/dygraph/train.py
+++ b/dygraph/train.py
@@ -121,7 +121,7 @@ def main(args):
         cfg.model,
         train_dataset,
         places=places,
-        eval_dataset=val_dataset,
+        val_dataset=val_dataset,
         optimizer=cfg.optimizer,
         save_dir=args.save_dir,
         iters=cfg.iters,
-- 
GitLab