diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 2c9c1f6c7d7a58aa295acc6a6270d1d522823996..da829282ae348546b85db8a3ef3b4bd97003a3f1 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 98d51de5c63a2d2fe917087090210ebd7f9a08b5..64e0e35d1d71e251bf381bfc671424838193e361 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