From d6dbb4b5b19d05b57feef9b01cfc4ddfc404107c Mon Sep 17 00:00:00 2001 From: Jeff Rasley Date: Wed, 18 Nov 2020 22:23:57 +0000 Subject: [PATCH] append job-name if explicit output dir is given --- deepspeed/runtime/engine.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/deepspeed/runtime/engine.py b/deepspeed/runtime/engine.py index b360e12b10b9..2322c0b10ba8 100755 --- a/deepspeed/runtime/engine.py +++ b/deepspeed/runtime/engine.py @@ -264,7 +264,9 @@ def get_summary_writer(self, base=os.path.join(os.environ["HOME"], "tensorboard")): if self.tensorboard_output_path(): - log_dir = self.tensorboard_output_path() + base_dir = self.tensorboard_output_path() + job_name = self.tensorboard_job_name() + log_dir = os.path.join(base_dir, job_name) else: if self.tensorboard_job_name(): name = self.tensorboard_job_name()