fix ldc2d_unsteady.md、ldc2d_unsteady_Re10.yaml、ldc2d_unsteady_Re10.py

docs/zh/examples/ldc2d_unsteady.md

@@ -4,12 +4,10 @@
 
 === "模型训练命令"
 ``` sh
+    python ldc2d_unsteady_Re10.py
 ```
 
 
-=== "模型评估命令"
-``` sh
-```
 
 ## 1. 背景简介
 

examples/ldc/conf/ldc2d_unsteady_Re10.yaml

@@ -5,6 +5,14 @@ hydra:
   job:
     name: ${mode} # name of logfile
     chdir: false # keep current working direcotry unchanged
+    config: 
+      override_dirname:
+        exclude_keys:
+          - TRAIN.checkpoint_path
+          - TRAIN.pretrained_model_path
+          - EVAL.pretrained_model_path
+          - mode
+          - output_dir
   sweep:
     # output directory for multirun
     dir: ${hydra.run.dir}

examples/ldc/ldc2d_unsteady_Re10.py

@@ -18,20 +18,23 @@
 from omegaconf import DictConfig
 
 import ppsci
-# from ppsci.utils import config
 from ppsci.utils import logger
 
 def train(cfg: DictConfig):
     # set random seed for reproducibility
     ppsci.utils.misc.set_random_seed(cfg.seed)
     # initialize logger
     logger.init_logger("ppsci", osp.join(cfg.output_dir, "train.log"), "info")
+
     # set model
     model = ppsci.arch.MLP(**cfg.MODEL.model)
+
     # set equation
     equation = {"NavierStokes": ppsci.equation.NavierStokes(0.01, 1.0, 2, True)}
+
     # set timestamps(including initial t0)
     timestamps = np.linspace(0.0, 1.5, 16, endpoint=True)
+
     # set time-geometry
     geom = {
         "time_rect": ppsci.geometry.TimeXGeometry(
@@ -122,7 +125,6 @@ def train(cfg: DictConfig):
     }
 
     # set training hyper-parameters
-
     lr_scheduler = ppsci.optimizer.lr_scheduler.Cosine(
         **cfg.TRAIN.lr_scheduler,
         warmup_epoch=int(0.05 * cfg.TRAIN.epochs),
@@ -213,19 +215,24 @@ def evaluate(cfg: DictConfig):
     ppsci.utils.misc.set_random_seed(cfg.seed)
     # initialize logger
     logger.init_logger("ppsci", osp.join(cfg.output_dir, "train.log"), "info")
+
     # set model
     model = ppsci.arch.MLP(**cfg.MODEL.model)
+
     # set equation
     equation = {"NavierStokes": ppsci.equation.NavierStokes(0.01, 1.0, 2, True)}
+
     # set timestamps(including initial t0)
     timestamps = np.linspace(0.0, 1.5, 16, endpoint=True)
+
     # set time-geometry
     geom = {
         "time_rect": ppsci.geometry.TimeXGeometry(
             ppsci.geometry.TimeDomain(0.0, 1.5, timestamps=timestamps),
             ppsci.geometry.Rectangle((-0.05, -0.05), (0.05, 0.05)),
         )
     }
+
     # pde/bc constraint use t1~tn, initial constraint use t0
     NTIME_ALL = len(timestamps)
     NPOINT_PDE, NTIME_PDE = 99**2, NTIME_ALL - 1
@@ -235,8 +242,6 @@ def evaluate(cfg: DictConfig):
     NPOINT_RIGHT, NTIME_RIGHT = 99, NTIME_ALL - 1
     NPOINT_IC, NTIME_IC = 99**2, 1
 
-
-
     # set validator
     NPOINT_EVAL = NPOINT_PDE * NTIME_ALL
     residual_validator = ppsci.validate.GeometryValidator(
@@ -256,6 +261,7 @@ def evaluate(cfg: DictConfig):
         name="Residual",
     )
     validator = {residual_validator.name: residual_validator}
+
     # set visualizer(optional)
     NPOINT_BC = NPOINT_TOP + NPOINT_DOWN + NPOINT_LEFT + NPOINT_RIGHT
     vis_initial_points = geom["time_rect"].sample_initial_interior(
@@ -288,16 +294,6 @@ def evaluate(cfg: DictConfig):
             prefix="result_u_v",
         )
     }
-    # set visualizer(optional)
-    NPOINT_BC = NPOINT_TOP + NPOINT_DOWN + NPOINT_LEFT + NPOINT_RIGHT
-    vis_initial_points = geom["time_rect"].sample_initial_interior(
-        (NPOINT_IC + NPOINT_BC), evenly=True
-    )
-    vis_pde_points = geom["time_rect"].sample_interior(
-        (NPOINT_PDE + NPOINT_BC) * NTIME_PDE, evenly=True
-    )
-    vis_points = vis_initial_points
-
 
     # directly evaluate pretrained model(optional)
     solver = ppsci.solver.Solver(
@@ -323,4 +319,4 @@ def main(cfg: DictConfig):
 
 
 if __name__ == "__main__":
-    main()
+    main()