pytorch 画loss曲线_Pytorch学习_visdom可视化(0.01):如何画Loss图片

画loss,但是一直遇到一个问题，定义窗口时，需要画第一个点(一般是原点)，但是这边后面增加点，导致append到后面，但是第一点没办法处理。

安装visdom

pip install visdom 或者 conda isntall visdom

打开

python -m visdom.server

#通过http://localhost:8097访问

使用visdom

from visdom import Visdom

viz = Visdom()

viz.line([0.], [0.], win='train_loss',opts=dict(title='train loss')) #single-line

viz.line([loss.item()], [global_step], win='trian_loss', update='append')

###################

viz.line([[0., 0.]], [0.], win='test',opts=dict(title='train loss',

legend=['loss', 'acc.'])) #single-line

viz.line([[test_loss, correct / len(test_loader.dataset)]], [global_step], win='test', update='append')

###################

viz.images(data.view(-1