Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "b2c63c79c3c7658c5783340dba9d049cf3824e5d"
Unverified Commit fe379f85 authored by Chengjiang Li's avatar Chengjiang Li Committed by GitHub
Browse files

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

parent d8049331
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/transformers/integrations.py
View file @ fe379f85
...@@ -364,6 +364,7 @@ class TensorBoardCallback(TrainerCallback): ...@@ -364,6 +364,7 @@ class TensorBoardCallback(TrainerCallback):
if trial_name is not None: if trial_name is not None:
log_dir = os.path.join(args.logging_dir, trial_name) log_dir = os.path.join(args.logging_dir, trial_name)
if self.tb_writer is None:
self._init_summary_writer(args, log_dir) self._init_summary_writer(args, log_dir)
if self.tb_writer is not None: if self.tb_writer is not None:
......
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