[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
Show 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,6 +364,7 @@ class TensorBoardCallback(TrainerCallback):
            if trial_name is not None:
                log_dir = os.path.join(args.logging_dir, trial_name)

+        if self.tb_writer is None:
            self._init_summary_writer(args, log_dir)

        if self.tb_writer is not None: