pytorch学习笔记6-visdom可视化工具的使用

1. 概述

在深度学习框架pytorch中，可视化工具尤为重要，它可以实时监督模型训练的状况，并可视化出来，相对于visdom有一种可视化工具叫tensorboardX，因为它大概30s才刷新一次，并且运行起来比较占用内存所以现在都在用visdom进行深度学习可视化的监督。下来介绍visdom的基本用法。

2. 安装

在命令行cmd或者终端terminal中运行下面的语句：

pip install visdom

即可完成visdom的安装

3. 启动visdom工具

输入命令

python -m visdom.server

启动成功就可以在控制面板看到 http://localhost:8097
点击进去就是visdom的面板了

4. visdom工具的使用

 viz = Visdom()
 viz.line([0.],[0.],win='train_loss',opts=dict(title='train loss'))
 viz.line([loss.item()],[global_step],win='train_loss',update='append')

 from visdom import Visdom
 viz = Visdom()
 viz.line([0.0],[0.0],win='test',opts=dict(title='test loss&acc.',legend=['loss','acc']))
 viz.line([[test_loss, correct / len(test_loader.dataset)]],[global_step],win='test',update='append')

 from visdom import Visdom
 viz = Visdom()
 viz.images(data.view(-1,1,28,28), win='x')
 viz.text(str(pred.detach().cpu().numpy()),win='pred',opts=dict(title='pred'))

5. 最终结果