1、安装Visdom包
首先在Anaconda命令行下激活pytorch环境,在pytorch环境下安装visdom所需要的包。如下图所示,为安装成功界面。
2、如何启动
由于visdom的运行需要终端服务器的支持,所以需要在当前命令行输入python -m visdom.server打开一个终端监听窗口,并将返回的网址复制到浏览器即可。
3、新建visdom窗口,实时显示loss函数的变化
from visdom import Visdom
if __name__ == '__main__':
wind = Visdom()
wind.line([0.], # Y的第一个点的坐标
[0.], # X的第一个点的坐标
win="loss", # 窗口的名称
opts=dict(title='loss'))
for epoch in range(1000):
for i,data in enumerate(train_loader,0):
inputs,labels = data
y_pred = model(inputs)
loss = criterion(y_pred,labels)
print(epoch,i,loss.item())
optimizer.zero_grad()
loss.backward()
optimizer.step()
wind.line([loss.item()],[epoch],win="loss",update='append')