使用visdom 制作随时间动态增长的图

安装visdom

pip install visdom

启动visdom

python -m visdom.server

随时间动态增长的visdom图

典型的例子如：创建追踪loss变化的图。

用法：update选项设置为“append”

举例

track_loss = 0 #for draw graph
global_step = 0
vis = visdom.Visdom(env=u"train_loss")
win = vis.line(X = np.array([global_step]), Y = np.array([track_loss]))

for epoch in range(cfg.epoch_num):
    # 此处省略代码

    for iter_num, dial_batch in enumerate(data_iterator):
        # 此处省略代码
        loss = some_function() 
        vis.line(X = np.array([global_step]), Y = np.array([loss.data[0]]), win = win,                
                 update = 'append')  # for draw graph
        global_step += 1

 

生成的visdom图像如下所示：

 

---