Merge pull request #3 from yihaosun1124/main

fix the tensorboard bug
polixir · Jun 3, 2022 · eee1912 · eee1912
2 parents 085fd71 + a9ffd2d
commit eee1912
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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