Merge pull request #2 from wassname/patch-1

create logger.py fix issue #1

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)