Pytorch中使用tensorboard中如何添加低维映射add_embedding
这篇文章给大家介绍Pytorch中使用tensorboard中如何添加低维映射add_embedding,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。
pytorch 中使用tensorboard,详解writer.add_embedding函数的作用
代码展示:
from torch.utils.tensorboard import SummaryWriterimport keywordimport torch writer = SummaryWriter()meta = []while len(meta)<100:meta = meta+keyword.kwlist # get some stringsmeta = meta[:100]for i, v in enumerate(meta):meta[i] = v+str(i)label_img = torch.rand(100, 3, 10, 32)for i in range(100):label_img[i]*=i/100.0writer.add_embedding(torch.randn(100, 5), metadata=meta, label_img=label_img)# writer.add_embedding(torch.randn(100, 5), label_img=label_img)# writer.add_embedding(torch.randn(100, 5), metadata=meta)writer.close()# tensorboard --logdir=runs
运行结果(浏览器的页面需要刷新一下):
关于Pytorch中使用tensorboard中如何添加低维映射add_embedding就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。