温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

Pytorch中使用tensorboard中如何添加低维映射add_embedding

发布时间:2021-12-04 18:55:28 来源:亿速云 阅读:427 作者:柒染 栏目:大数据

这篇文章给大家介绍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

关于Pytorch中使用tensorboard中如何添加低维映射add_embedding就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI