from tensorboard import SummaryWriter failed in train.py

[rongzha1]

from tensorboard import SummaryWriter Traceback (most recent call last): File "", line 1, in ImportError: cannot import name SummaryWriter

tensorboard (1.6.0) failed import SummaryWriter I use tensorboardX (1.1) instead, and changed it to from tensorboardX import SummaryWriter. It works well.

[Soonhwan-Kwon]

Thank you for finding this error, we had no time to confirm it until now, you helps our project very much, thank you! I can also confirmed it today, again thank you for reporting us and we will fix it soon.

[xinyu-intel]

use tensorboardX and change to from tensorboardX import SummaryWriter will fix this bug.