visdom

作者: blair_liu | 来源:发表于2021-01-25 14:16 被阅读0次
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')

相关文章

网友评论

      本文标题:visdom

      本文链接:https://www.haomeiwen.com/subject/nfabzktx.html