add exclude_keys for amgnet yaml and fix document typo (PaddlePaddle#677

)
HydrogenSulfate · Dec 4, 2023 · 6b3cbf3 · 6b3cbf3
1 parent 842ac9f
commit 6b3cbf3
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 1 deletion.
diff --git a/docs/zh/examples/amgnet.md b/docs/zh/examples/amgnet.md
@@ -38,7 +38,7 @@
         # curl https://paddle-org.bj.bcebos.com/paddlescience/datasets/AMGNet/data.zip --output data.zip
         # unzip it
         unzip data.zip
-        python amgnet_airfoil.py mode=eval EVAL.pretained_model_path=https://paddle-org.bj.bcebos.com/paddlescience/models/amgnet/amgnet_airfoil_pretrained.pdparams
+        python amgnet_airfoil.py mode=eval EVAL.pretrained_model_path=https://paddle-org.bj.bcebos.com/paddlescience/models/amgnet/amgnet_airfoil_pretrained.pdparams
         ```
     === "amgnet_cylinder"
 

diff --git a/examples/amgnet/amgnet_airfoil.py b/examples/amgnet/amgnet_airfoil.py
@@ -152,6 +152,11 @@ def train(cfg: DictConfig):
 
 
 def evaluate(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, "eval.log"), "info")
+
     # set airfoil model
     model = ppsci.arch.AMGNet(**cfg.MODEL)
 

diff --git a/examples/amgnet/amgnet_cylinder.py b/examples/amgnet/amgnet_cylinder.py
@@ -152,6 +152,11 @@ def train(cfg: DictConfig):
 
 
 def evaluate(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, "eval.log"), "info")
+
     # set airfoil model
     model = ppsci.arch.AMGNet(**cfg.MODEL)
 

diff --git a/examples/amgnet/conf/amgnet_airfoil.yaml b/examples/amgnet/conf/amgnet_airfoil.yaml
@@ -5,6 +5,15 @@ hydra:
   job:
     name: ${mode} # name of logfile
     chdir: false # keep current working direcotry unchaned
+    config:
+      override_dirname:
+        exclude_keys:
+          - TRAIN.checkpoint_path
+          - TRAIN.pretrained_model_path
+          - EVAL.pretrained_model_path
+          - mode
+          - output_dir
+          - log_freq
   sweep:
     # output directory for multirun
     dir: ${hydra.run.dir}

diff --git a/examples/amgnet/conf/amgnet_cylinder.yaml b/examples/amgnet/conf/amgnet_cylinder.yaml
@@ -5,6 +5,15 @@ hydra:
   job:
     name: ${mode} # name of logfile
     chdir: false # keep current working direcotry unchaned
+    config:
+      override_dirname:
+        exclude_keys:
+          - TRAIN.checkpoint_path
+          - TRAIN.pretrained_model_path
+          - EVAL.pretrained_model_path
+          - mode
+          - output_dir
+          - log_freq
   sweep:
     # output directory for multirun
     dir: ${hydra.run.dir}