Skip to content

GitLab

Unverified Commit 24a66f1d authored by NightBaronStar's avatar NightBaronStar Committed by GitHub
Browse files

Update tensorboard.py (#205) 

* Update tensorboard.py

* Update tensorboard.py
parent fd066152
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmcv/runner/hooks/logger/tensorboard.py
View file @ 24a66f1d
...@@ -22,7 +22,7 @@ class TensorboardLoggerHook(LoggerHook): ...@@ -22,7 +22,7 @@ class TensorboardLoggerHook(LoggerHook):
@master_only @master_only
def before_run(self, runner): def before_run(self, runner):
if torch.__version__ >= '1.1': if torch.__version__ >= '1.1' and '.' in torch.__version__:
try: try:
from torch.utils.tensorboard import SummaryWriter from torch.utils.tensorboard import SummaryWriter
except ImportError: except ImportError:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment