append job-name if explicit output dir is given (#539)

microsoft · Nov 18, 2020 · 5b09be6 · 5b09be6
1 parent 7752dc5
commit 5b09be6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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()