[Fix] Fix output_dir for visualDL and tensorboardX

docs/zh/user_guide.md

@@ -112,7 +112,7 @@ TRAIN:
 ``` sh title="$ python bracket.py {++-m seed=42,1024 TRAIN.epochs=10,20++}"
 [HYDRA] Launching 4 jobs locally
 [HYDRA]        #0 : seed=42 TRAIN.epochs=10
-....
+...
 [HYDRA]        #1 : seed=42 TRAIN.epochs=20
 ...
 [HYDRA]        #2 : seed=1024 TRAIN.epochs=10
@@ -696,7 +696,7 @@ hydra 的自动化实验功能可以与 [optuna](https://optuna.readthedocs.io/e
 
 2. 修改 `viv.yaml` 文件，在 `defaults:` 和 `hydra:` 字段下分别添加如下配置（高亮部分所示）
 
-    ``` yaml title="title" hl_lines="8 26-34"
+    ``` yaml title="viv.yaml" hl_lines="8 26-34"
     defaults:
       - ppsci_default
       - TRAIN: train_default

ppsci/solver/solver.py

@@ -380,7 +380,7 @@ def dist_wrapper(model: nn.Layer) -> paddle.DataParallel:
         if self.use_vdl:
             with misc.RankZeroOnly(self.rank) as is_master:
                 if is_master:
-                    self.vdl_writer = vdl.LogWriter(osp.join(output_dir, "vdl"))
+                    self.vdl_writer = vdl.LogWriter(osp.join(self.output_dir, "vdl"))
             logger.info(
                 "VisualDL is enabled for logging, you can view it by "
                 f"running:\nvisualdl --logdir {self.vdl_writer._logdir} --port 8080"
@@ -415,7 +415,7 @@ def dist_wrapper(model: nn.Layer) -> paddle.DataParallel:
             with misc.RankZeroOnly(self.rank) as is_master:
                 if is_master:
                     self.tbd_writer = tensorboardX.SummaryWriter(
-                        osp.join(output_dir, "tensorboard")
+                        osp.join(self.output_dir, "tensorboard")
                     )
             logger.message(
                 "TensorboardX is enabled for logging, you can view it by "

requirements.txt

@@ -9,6 +9,7 @@ pydantic>=2.5.0
 pyevtk
 pyvista==0.37.0
 pyyaml
+scikit-learn<1.5.0
 scikit-optimize
 scipy
 seaborn