[Fix]Make sure the args tb_writer passed to the TensorBoardCallback works (#13636)

fe379f85 · Chengjiang Li · GitHub · d8049331 · fe379f85
Unverified Commit fe379f85 authored Sep 20, 2021 by Chengjiang Li Committed by GitHub Sep 20, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/transformers/integrations.py src/transformers/integrations.py +2 -1

No files found.
--- a/src/transformers/integrations.py
+++ b/src/transformers/integrations.py
@@ -364,7 +364,8 @@ class TensorBoardCallback(TrainerCallback):
            if trial_name is not None:
                log_dir = os.path.join(args.logging_dir, trial_name)
-        self._init_summary_writer(args, log_dir)
+        if self.tb_writer is None:
+            self._init_summary_writer(args, log_dir)
        if self.tb_writer is not None:
            self.tb_writer.add_text("args", args.to_json_string())