Support python3 and pytorch 0.2

utils/__init__.py

@@ -1,7 +1,8 @@
-from config import *
-from normalizer import *
-from run import *
+from .config import *
+from .normalizer import *
+from .run import *
+
 try:
-    from tf_logger import Logger
+    from .tf_logger import Logger
 except:
-    from vanilla_logger import Logger
+    from .vanilla_logger import Logger

utils/tf_logger.py

@@ -3,6 +3,7 @@
 import tensorflow as tf
 import numpy as np
 import scipy.misc
+import logging
 
 try:
     from StringIO import StringIO  # Python 2.7
@@ -18,6 +19,7 @@ class Logger(object):
         self.debug = vanilla_logger.debug
         self.warning = vanilla_logger.warning
         self.skip = skip
+        logging.info('')
 
     def scalar_summary(self, tag, value, step):
         if self.skip:

utils/vanilla_logger.py

@@ -1,4 +1,5 @@
 import numpy as np
+import logging
 
 class Logger(object):
     def __init__(self, log_dir, vanilla_logger, skip=False):
@@ -7,6 +8,7 @@ class Logger(object):
         self.debug = vanilla_logger.debug
         self.warning = vanilla_logger.warning
         self.skip = skip
+        logging.info('')
 
     def scalar_summary(self, tag, value, step):
         if self.skip: