Skip to content

Commit 6099dcc

Browse files
[Fix] Fix output_dir for visualDL and tensorboardX (#908)
* fix output_dir for visualDL and tensorboardX * update scikit-learn<1.5.0 for incompatible upgrad
1 parent 7168b9a commit 6099dcc

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

docs/zh/user_guide.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ TRAIN:
112112
``` sh title="$ python bracket.py {++-m seed=42,1024 TRAIN.epochs=10,20++}"
113113
[HYDRA] Launching 4 jobs locally
114114
[HYDRA] #0 : seed=42 TRAIN.epochs=10
115-
....
115+
...
116116
[HYDRA] #1 : seed=42 TRAIN.epochs=20
117117
...
118118
[HYDRA] #2 : seed=1024 TRAIN.epochs=10
@@ -696,7 +696,7 @@ hydra 的自动化实验功能可以与 [optuna](https://optuna.readthedocs.io/e
696696
697697
2. 修改 `viv.yaml` 文件,在 `defaults:` 和 `hydra:` 字段下分别添加如下配置(高亮部分所示)
698698
699-
``` yaml title="title" hl_lines="8 26-34"
699+
``` yaml title="viv.yaml" hl_lines="8 26-34"
700700
defaults:
701701
- ppsci_default
702702
- TRAIN: train_default

ppsci/solver/solver.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -380,7 +380,7 @@ def dist_wrapper(model: nn.Layer) -> paddle.DataParallel:
380380
if self.use_vdl:
381381
with misc.RankZeroOnly(self.rank) as is_master:
382382
if is_master:
383-
self.vdl_writer = vdl.LogWriter(osp.join(output_dir, "vdl"))
383+
self.vdl_writer = vdl.LogWriter(osp.join(self.output_dir, "vdl"))
384384
logger.info(
385385
"VisualDL is enabled for logging, you can view it by "
386386
f"running:\nvisualdl --logdir {self.vdl_writer._logdir} --port 8080"
@@ -415,7 +415,7 @@ def dist_wrapper(model: nn.Layer) -> paddle.DataParallel:
415415
with misc.RankZeroOnly(self.rank) as is_master:
416416
if is_master:
417417
self.tbd_writer = tensorboardX.SummaryWriter(
418-
osp.join(output_dir, "tensorboard")
418+
osp.join(self.output_dir, "tensorboard")
419419
)
420420
logger.message(
421421
"TensorboardX is enabled for logging, you can view it by "

requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ pydantic>=2.5.0
99
pyevtk
1010
pyvista==0.37.0
1111
pyyaml
12+
scikit-learn<1.5.0
1213
scikit-optimize
1314
scipy
1415
seaborn

0 commit comments

Comments
 (0)