test loss可视化 pytorch

Pytorch可视化： https://zhuanlan.zhihu.com/p/147158851
github官方介绍： https://github.com/noagarcia/visdom-tutorial

from visdom import Visdom

viz = Visdom()

train_global = 0

test_global = 0

train_global += 1

viz.line([loss.item()], [train_global], win='train_loss', update="append")

if (epoch % 1 == 0):

{