SummaryWriter 有时引用torch的报错,解决方法:
# from torch.utils.tensorboard import SummaryWriter
from tensorboardX import SummaryWriter
SummaryWriter 有时引用torch的报错,解决方法:
# from torch.utils.tensorboard import SummaryWriter
from tensorboardX import SummaryWriter