"tests/encoder_decoder/test_modeling_encoder_decoder.py" did not exist on "5ec368d79ecfa1467358f5b42da2cecbe23d6b0d"
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
...@@ -364,7 +364,8 @@ class TensorBoardCallback(TrainerCallback): ...@@ -364,7 +364,8 @@ 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)
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: if self.tb_writer is not None:
self.tb_writer.add_text("args", args.to_json_string()) self.tb_writer.add_text("args", args.to_json_string())
......
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