replace home env with ~

49496364 · Jeff Rasley · GitHub · 6adc19a6 · 49496364
Unverified Commit 49496364 authored Mar 09, 2021 by Jeff Rasley Committed by GitHub Mar 09, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

deepspeed/runtime/engine.py deepspeed/runtime/engine.py +1 -1

No files found.
--- a/deepspeed/runtime/engine.py
+++ b/deepspeed/runtime/engine.py
@@ -255,7 +255,7 @@ class DeepSpeedEngine(Module):

    def get_summary_writer(self,
                           name="DeepSpeedJobName",
-                           base=os.path.join(os.environ["HOME"],
+                           base=os.path.join(os.path.expanduser("~"),
                                             "tensorboard")):
        if self.tensorboard_output_path():
            base_dir = self.tensorboard_output_path()