Fix eval config (#948)

Author: HydrogenSulfate

examples/aneurysm/aneurysm.py

@@ -190,7 +190,7 @@ def inlet_w_ref_func(_in):
         "num_workers": 1,
     }
     sup_validator = ppsci.validate.SupervisedValidator(
-        {**eval_dataloader_cfg, "batch_size": cfg.EVAL.batch_size.sup_validator},
+        {**eval_dataloader_cfg, "batch_size": cfg.EVAL.batch_size},
         ppsci.loss.MSELoss("mean"),
         {
             "p": lambda out: out["p"],
@@ -213,7 +213,7 @@ def inlet_w_ref_func(_in):
                 "v": lambda out: out["v"],
                 "w": lambda out: out["w"],
             },
-            batch_size=cfg.EVAL.batch_size.sup_validator,
+            batch_size=cfg.EVAL.batch_size,
             prefix="result_u_v_w_p",
         ),
     }
@@ -288,7 +288,7 @@ def evaluate(cfg: DictConfig):
         "num_workers": 1,
     }
     sup_validator = ppsci.validate.SupervisedValidator(
-        {**eval_dataloader_cfg, "batch_size": cfg.EVAL.batch_size.sup_validator},
+        {**eval_dataloader_cfg, "batch_size": cfg.EVAL.batch_size},
         ppsci.loss.MSELoss("mean"),
         {
             "p": lambda out: out["p"],
@@ -311,7 +311,7 @@ def evaluate(cfg: DictConfig):
                 "v": lambda out: out["v"],
                 "w": lambda out: out["w"],
             },
-            batch_size=cfg.EVAL.batch_size.sup_validator,
+            batch_size=cfg.EVAL.batch_size,
             prefix="result_u_v_w_p",
         ),
     }

examples/aneurysm/conf/aneurysm.yaml

@@ -97,8 +97,7 @@ TRAIN:
 EVAL:
   pretrained_model_path: null
   eval_with_no_grad: true
-  batch_size:
-    sup_validator: 4096
+  batch_size: 4096
 
 # inference settings
 INFER:

examples/biharmonic2d/conf/biharmonic2d.yaml

@@ -72,8 +72,7 @@ TRAIN:
 EVAL:
   pretrained_model_path: null
   eval_with_no_grad: true
-  batch_size:
-    sup_validator: 128
+  batch_size: 128
 
 INFER:
   pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/biharmonic2d/biharmonic2d_pretrained.pdparams

examples/bracket/bracket.py

@@ -284,7 +284,7 @@ def train(cfg: DictConfig):
         },
     }
     sup_validator = ppsci.validate.SupervisedValidator(
-        {**eval_dataloader_cfg, "batch_size": cfg.EVAL.batch_size.sup_validator},
+        {**eval_dataloader_cfg, "batch_size": cfg.EVAL.batch_size},
         ppsci.loss.MSELoss("mean"),
         {
             "u": lambda out: out["u"],
@@ -460,7 +460,7 @@ def evaluate(cfg: DictConfig):
         },
     }
     sup_validator = ppsci.validate.SupervisedValidator(
-        {**eval_dataloader_cfg, "batch_size": cfg.EVAL.batch_size.sup_validator},
+        {**eval_dataloader_cfg, "batch_size": cfg.EVAL.batch_size},
         ppsci.loss.MSELoss("mean"),
         {
             "u": lambda out: out["u"],

examples/bracket/conf/bracket.yaml

@@ -100,8 +100,7 @@ TRAIN:
 EVAL:
   pretrained_model_path: null
   eval_with_no_grad: true
-  batch_size:
-    sup_validator: 128
+  batch_size: 128
 
 # inference settings
 INFER:

examples/fpde/conf/fractional_poisson_2d.yaml

@@ -65,5 +65,4 @@ TRAIN:
 EVAL:
   pretrained_model_path: null
   eval_with_no_grad: true
-  batch_size:
-    sup_validator: 128
+  batch_size: 128