import visdom
vis = visdom.Visdom(env='main')
if epoch == 0:
vis.line([[train_epoch_loss, validation_epoch_loss]], [epoch], win='train', opts=dict(title='train_loss&val_loss', legend=['train_loss', 'val_loss']))
else:
vis.line([[train_epoch_loss, validation_epoch_loss]], [epoch], win='train', update='append')
网友评论