From de5c4e1b1cec4349760c1f8daa2f789680e15825 Mon Sep 17 00:00:00 2001 From: Bobholamovic Date: Tue, 13 Jun 2023 23:01:48 +0800 Subject: [PATCH] Change vdl dir --- ppcls/engine/engine.py | 2 +- ppcls/static/train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 2c9c1f6c..da829282 100755 --- a/ppcls/engine/engine.py +++ b/ppcls/engine/engine.py @@ -94,7 +94,7 @@ class Engine(object): self.vdl_writer = None if self.config['Global'][ 'use_visualdl'] and mode == "train" and dist.get_rank() == 0: - vdl_writer_path = os.path.join(self.output_dir, "vdl") + vdl_writer_path = self.output_dir if not os.path.exists(vdl_writer_path): os.makedirs(vdl_writer_path) self.vdl_writer = LogWriter(logdir=vdl_writer_path) diff --git a/ppcls/static/train.py b/ppcls/static/train.py index 98d51de5..64e0e35d 100755 --- a/ppcls/static/train.py +++ b/ppcls/static/train.py @@ -110,7 +110,7 @@ def main(args): # visualDL vdl_writer = None if global_config["use_visualdl"]: - vdl_dir = os.path.join(global_config["output_dir"], "vdl") + vdl_dir = global_config["output_dir"] vdl_writer = LogWriter(vdl_dir) # build dataloader -- GitLab