一、Visdom
1、Visdom是Pytorch的一个可视化工具:支持数值、图像、文本、视频、折线图等格式
2、安装:
pip install visdom
3、启动
python -m visdom.server
二、tensorboardX
1、安装:
pip install tensorboardX
2、案例:
from tensorboardX import SummaryWriter
w = SummaryWriter('./log')
for i in range(10):
w.add_scalar('a', i, global_step=i)
w.add_scalar('b', i**2, global_step=i)
w.close()