From 8730f9c33f52bcb47dc9bb42024d9aa43c66979f Mon Sep 17 00:00:00 2001
From: Mike Clark <wassname@users.noreply.github.com>
Date: Sat, 20 Jan 2018 11:58:01 +0800
Subject: [PATCH] create logger.py fix issue #1

Fixes issue #1
---
 common/logger.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 common/logger.py

diff --git a/common/logger.py b/common/logger.py
new file mode 100644
index 0000000..d990516
--- /dev/null
+++ b/common/logger.py
@@ -0,0 +1,22 @@
+from tensorboardX import SummaryWriter
+import logging
+
+logger = logging.getLogger(__name__)
+
+class Logger(object):
+    def __init__(self, log_dir, vanilla_logger=logger, skip=False):
+        self.writer = SummaryWriter(log_dir)
+        self.info = vanilla_logger.info
+        self.debug = vanilla_logger.debug
+        self.warning = vanilla_logger.warning
+        self.skip = skip
+
+    def scalar_summary(self, tag, value, step):
+        if self.skip:
+            return
+        self.writer.add_scalar(tag, value, step)
+
+    def histo_summary(self, tag, values, step):
+        if self.skip:
+            return
+        self.writer.add_histogram(tag, values, step, bins=1000)