diff --git a/docs/zh/user_guide.md b/docs/zh/user_guide.md index 19cf41f783..59ead15aa4 100644 --- a/docs/zh/user_guide.md +++ b/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 diff --git a/ppsci/solver/solver.py b/ppsci/solver/solver.py index dc861a0e7d..8fb27928ae 100644 --- a/ppsci/solver/solver.py +++ b/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 " diff --git a/requirements.txt b/requirements.txt index aeb4da23b2..2c1594d67a 100644 --- a/requirements.txt +++ b/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