https://blog.csdn.net/bigbennyguo/article/details/87956434
from torch.utils.tensorborad import SummaryWriter
from tensorboardX import SummaryWriter
Pytorch训练可视化
原文：https://www.cnblogs.com/JunzhaoLiang/p/12182616.html
