安装:
pip install visdom
pip install torchnet(为了使用meter,轻量级管理损失,精度的工具)
使用:
先打开visdom
python -m visdom.server
通过http://localhost:8097访问
代码示例:
from visualize import Visualizer
#对visdom的封装 源码https://github.com/chenyuntc/pytorch-best-practice/blob/master/utils/visualize.py
from torchnet import meter
vis = Visualizer(env='baseGAN')
g_loss_meter = meter.AverageValueMeter()
d_loss_meter = meter.AverageValueMeter()
g_loss_meter.add(g_loss.data[0])
d_loss_meter.add(d_loss.data[0])
vis.plot_many({'g_loss': g_loss_meter.value()[0], 'd_loss': d_loss_meter.value()[0]})