PyTorch用TensorboardX实现训练可视化

1.常用语句如下，在写好的代码的对应位置加上这几句话等类似的，就能显示出对应的图啊表啊之类的统计了

from tensorboardX import SummaryWriter

writer = SummaryWriter(log_dir='sg') #名称自己定

writer.add_scalar('Train', loss, epoch)

writer.add_scalars('Test',{'loss_g': loss_g, 'loss_d': loss_d}, i)

writer.add_graph(self.net.map_net, (latent))

效果如下图

 

2.  https://github.com/lanpa/tensorboardX/blob/master/examples/demo_graph.py 官方demo，重点理解下add_graph()里，当需要传参不唯一的时候，input_to_model的传参方式：

 

 

 

注：https://blog.csdn.net/bigbennyguo/article/details/87956434#_scalar_52  几种常用add介绍，陌生的可以先看一下