Visdom-pytorch-loss function and train_acc可视化

from visdom import Visdom 

line = viz.line(np.arange(10))
text = viz.text("<h1>convolution Nueral Network</h1>")
time_p, tr_acc, ts_acc, loss_p = [], [], [], []
time_p.append(epoch)
    tr_acc.append(train_acc)
    # ts_acc.append(acc)
    loss_p.append(train_loss)
   
    viz.line(X=np.column_stack((np.array(time_p), np.array(time_p))),
             Y=np.column_stack((np.array(loss_p), np.array(tr_acc))),
             win=line,
             opts=dict(legend=["Loss", "TRAIN_acc"]))
    # visdom text 支持html语句
     viz.text("<p style='color:red'>epoch:{}</p><br><p style='color:blue'>Loss:{:.4f}</p><br>".format(epoch, train_loss),win=text)
    

运行前先打开环境终端
输入 python -m visdom.server
复制链接与Google