Update tensorboard.py (#205)

* Update tensorboard.py * Update tensorboard.py
2020-03-11 17:01:52 +08:00 · 2020-03-11 17:01:52 +08:00 · 24a66f1d47
parent fd066152d0
commit 24a66f1d47
1 changed files with 1 additions and 1 deletions
--- a/mmcv/runner/hooks/logger/tensorboard.py
+++ b/mmcv/runner/hooks/logger/tensorboard.py
@ -22,7 +22,7 @@ class TensorboardLoggerHook(LoggerHook):

    @master_only
    def before_run(self, runner):
-        if torch.__version__ >= '1.1':
+        if torch.__version__ >= '1.1' and '.' in torch.__version__:
            try:
                from torch.utils.tensorboard import SummaryWriter
            except ImportError: