log meta (#183)

* log meta

* add type check

mmcv/runner/hooks/logger/text.py

@@ -21,6 +21,8 @@ class TextLoggerHook(LoggerHook):
         self.start_iter = runner.iter
         self.json_log_path = osp.join(runner.work_dir,
                                       '{}.log.json'.format(runner.timestamp))
+        if runner.meta is not None:
+            self._dump_log(runner.meta, runner)
 
     def _get_max_memory(self, runner):
         mem = torch.cuda.max_memory_allocated()

mmcv/runner/runner.py

@@ -31,6 +31,8 @@ class Runner(object):
         log_level (int): Logging level.
         logger (:obj:`logging.Logger`): Custom logger. If `None`, use the
             default logger.
+        meta (dict | None): A dict records some import information such as
+            environment info and seed, which will be logged in logger hook.
     """
 
     def __init__(self,
@@ -39,7 +41,8 @@ class Runner(object):
                  optimizer=None,
                  work_dir=None,
                  log_level=logging.INFO,
-                 logger=None):
+                 logger=None,
+                 meta=None):
         assert callable(batch_processor)
         self.model = model
         if optimizer is not None:
@@ -71,6 +74,10 @@ class Runner(object):
             self.logger = logger
         self.log_buffer = LogBuffer()
 
+        if meta is not None:
+            assert isinstance(meta, dict), '"meta" must be a dict or None'
+        self.meta = meta
+
         self.mode = None
         self._hooks = []
         self._epoch = 0