From a9ffd2d7de66bd48796cf752e3c7062c3b246dab Mon Sep 17 00:00:00 2001 From: Yihao-Sun <1778826780@qq.com> Date: Fri, 3 Jun 2022 19:19:14 +0800 Subject: [PATCH] fix the tensorboard bug --- RLA/easy_log/logger.py | 2 +- RLA/easy_log/tester.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/RLA/easy_log/logger.py b/RLA/easy_log/logger.py index 6ac17c2..7f6ed4d 100644 --- a/RLA/easy_log/logger.py +++ b/RLA/easy_log/logger.py @@ -688,7 +688,7 @@ def configure(dir=None, format_strs=None, comm=None, framework='tensorflow'): if format_strs is None: format_strs = os.getenv('OPENAI_LOG_FORMAT', 'stdout,log,csv').split(',') format_strs = filter(None, format_strs) - output_formats = [make_output_format(f, dir, log_suffix) for f in format_strs] + output_formats = [make_output_format(f, dir, log_suffix, framework) for f in format_strs] warn_output_formats = make_output_format('warn', dir, log_suffix, framework) backup_output_formats = make_output_format('backup', dir, log_suffix, framework) diff --git a/RLA/easy_log/tester.py b/RLA/easy_log/tester.py index 9fce214..e7ba846 100644 --- a/RLA/easy_log/tester.py +++ b/RLA/easy_log/tester.py @@ -187,7 +187,7 @@ def _init_logger(self): self.writer = None # logger configure logger.info("store file %s" % self.pkl_file) - logger.configure(self.log_dir, self.private_config["LOG_USED"]) + logger.configure(self.log_dir, self.private_config["LOG_USED"], framework=self.private_config["DL_FRAMEWORK"]) for fmt in logger.Logger.CURRENT.output_formats: if isinstance(fmt, logger.TensorBoardOutputFormat): self.writer = fmt.writer