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')
上一篇 下一篇

猜你喜欢

热点阅读