Fix Deadlock fix (#256)

* Fix Deadlock fix

Resolves #255

* Revert deadlock fix and fix of deadlock fix

This reverts commit af85591fc7f226623bc9ebef93b76653bfac3438.

mmcv/runner/hooks/logger/text.py

@@ -124,16 +124,13 @@ class TextLoggerHook(LoggerHook):
         log_dict['iter'] = runner.inner_iter + 1
         # only record lr of the first param group
         log_dict['lr'] = runner.current_lr()[0]
-        memory = None
-        if torch.cuda.is_available():
-            memory = self._get_max_memory(runner)
         if mode == 'train':
             log_dict['time'] = runner.log_buffer.output['time']
             log_dict['data_time'] = runner.log_buffer.output['data_time']
 
             # statistic memory
-            if memory is not None:
-                log_dict['memory'] = memory
+            if torch.cuda.is_available():
+                log_dict['memory'] = self._get_max_memory(runner)
         for name, val in runner.log_buffer.output.items():
             if name in ['time', 'data_time']:
                 continue