Adding logger

logger.py

@@ -7,6 +7,7 @@ import torch
 import torchvision
 import numpy as np
 from termcolor import colored
+from datetime import datetime
 
 FORMAT_CONFIG = {
     'rl': {
@@ -93,8 +94,10 @@ class MetersGroup(object):
 class Logger(object):
     def __init__(self, log_dir, use_tb=True, config='rl'):
         self._log_dir = log_dir
+        now = datetime.now()
+        dt_string = now.strftime("%d_%m_%Y-%H_%M_%S")
         if use_tb:
-            tb_dir = os.path.join(log_dir, 'tb')
+            tb_dir = os.path.join(log_dir, 'runs/tb_'+dt_string)
             if os.path.exists(tb_dir):
                 shutil.rmtree(tb_dir)
             self._sw = SummaryWriter(tb_dir)