visdom
2021-01-25 本文已影响0人
blair_liu
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')