PyTorch可视化工具Visdom教程

什么是visdom

visdom是Facebook为PyTorch开发可视化工具,支持numpy和tensor，功能和Tensorboard差不多，在三维空间数据展示十分出色，其次可以远程访问，随时随地观察训练效果。

安装和启动

安装

pip install visdom             # 安装

启动

python -m visdom.server  # 打开cmd中的链接即可访问

• port ：运行服务器的端口，默认8097。
• base_url ：基本服务器 URL（默认 = /）。
• env_path ：要重新加载的序列化会话的路径。
• readonly : 标志以只读模式启动服务器。
• enable_login : 为服务器设置身份验证的标志，需要用户名和密码才能登录。
• bind_local : 标志使服务器只能从本地主机访问。

可视化

折线图Line

import numpy as np
import visdom
vis=visdom.Visdom(env='new_environment')  # 创建新环境

win = vis.line(
    X=np.column_stack((       # x 值
        np.arange(0, 10),
        np.arange(0, 10),
        np.arange(0, 10),
    )),
    Y=X**2+[10,20,30],         # y值
    opts=dict(
        dash = np.array(['solid', 'dash', 'dashdot']),     # 实线 虚线 长短点线
        linecolor = np.array([                             # 三条线的RGB颜色
            [50, 131, 125],
            [30, 61, 200],
            [25, 239, 128],
        ]),
        title =  'Acc',        # 标题
        xlabel = 'x轴',        # x 轴
        ylabel = 'y轴',        # y轴
    )
)

for x1 in range(20,40):
    y1 = np.random.rand() * 3 + 5
    vis.line([y1], [x1], win = 'loss', update = 'append')     # 追加新的数据
    time.sleep(0.5)

其中update：

• append: 追加新的点
• insert: 插入新的点
• update: 修改当前点
• remove: 删除当前点