Pytorch训练可视化

https://blog.csdn.net/bigbennyguo/article/details/87956434

from torch.utils.tensorborad import SummaryWriter

from tensorboardX import SummaryWriter