Update tensorboard.py (#205)

* Update tensorboard.py * Update tensorboard.py

Update tensorboard.py (#205)
* Update tensorboard.py * Update tensorboard.py
24a66f1d · NightBaronStar · GitHub · fd066152 · 24a66f1d
Unverified Commit 24a66f1d authored Mar 11, 2020 by NightBaronStar Committed by GitHub Mar 11, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmcv/runner/hooks/logger/tensorboard.py mmcv/runner/hooks/logger/tensorboard.py +1 -1

No files found.
--- 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: