Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "f93eb8ff5222b4f95d0f030e0f7813de0455ae4e"
Unverified Commit 5b09be60 authored by Jeff Rasley's avatar Jeff Rasley Committed by GitHub
Browse files

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

parent 7752dc5e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
deepspeed/runtime/engine.py
View file @ 5b09be60
...@@ -264,7 +264,9 @@ class DeepSpeedEngine(Module): ...@@ -264,7 +264,9 @@ class DeepSpeedEngine(Module):
base=os.path.join(os.environ["HOME"], base=os.path.join(os.environ["HOME"],
"tensorboard")): "tensorboard")):
if self.tensorboard_output_path(): 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: else:
if self.tensorboard_job_name(): if self.tensorboard_job_name():
name = self.tensorboard_job_name() name = self.tensorboard_job_name()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment