{"id":2451,"date":"2024-10-30T13:44:26","date_gmt":"2024-10-30T05:44:26","guid":{"rendered":"https:\/\/sanlangcode.com\/?p=2451"},"modified":"2024-10-30T13:44:26","modified_gmt":"2024-10-30T05:44:26","slug":"tensorboard%e9%87%8f%e5%8c%96%e8%af%84%e4%bb%b7pytorch%e8%ae%ad%e7%bb%83%e6%a8%a1%e5%9e%8b","status":"publish","type":"post","link":"https:\/\/sanlangcode.com\/index.php\/2024\/10\/30\/tensorboard%e9%87%8f%e5%8c%96%e8%af%84%e4%bb%b7pytorch%e8%ae%ad%e7%bb%83%e6%a8%a1%e5%9e%8b\/","title":{"rendered":"TensorBoard\u91cf\u5316\u8bc4\u4ef7Pytorch\u8bad\u7ec3\u6a21\u578b"},"content":{"rendered":"\n<p><code>torch.utils.tensorboard<\/code> \u662f PyTorch \u4e2d\u7528\u4e8e\u96c6\u6210 TensorBoard \u7684\u4e00\u4e2a\u6a21\u5757\uff0c\u65b9\u4fbf\u7528\u6237\u5728\u8bad\u7ec3\u6a21\u578b\u65f6\u8bb0\u5f55\u548c\u53ef\u89c6\u5316\u5404\u79cd\u6307\u6807\u3002TensorBoard \u662f\u4e00\u4e2a\u5f3a\u5927\u7684\u53ef\u89c6\u5316\u5de5\u5177\uff0c\u5e38\u7528\u4e8e\u76d1\u63a7\u548c\u8c03\u8bd5\u673a\u5668\u5b66\u4e60\u6a21\u578b\u7684\u8bad\u7ec3\u8fc7\u7a0b\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">\u4e3b\u8981\u529f\u80fd<\/h3>\n\n\n\n<ol class=\"wp-block-list\"><li><strong>\u6807\u91cf\u8bb0\u5f55<\/strong>\uff1a\u53ef\u4ee5\u8bb0\u5f55\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u7684\u6807\u91cf\u6570\u636e\uff0c\u5982\u635f\u5931\u503c\u548c\u51c6\u786e\u7387\u3002\u8fd9\u4e9b\u6570\u636e\u53ef\u4ee5\u5728 TensorBoard \u4e2d\u4ee5\u56fe\u8868\u7684\u5f62\u5f0f\u5448\u73b0\uff0c\u5e2e\u52a9\u5206\u6790\u6a21\u578b\u7684\u8bad\u7ec3\u6548\u679c\u3002<\/li><li><strong>\u56fe\u50cf\u8bb0\u5f55<\/strong>\uff1a\u652f\u6301\u8bb0\u5f55\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u751f\u6210\u7684\u56fe\u50cf\uff0c\u53ef\u4ee5\u7528\u4e8e\u53ef\u89c6\u5316\u6a21\u578b\u751f\u6210\u7684\u8f93\u51fa\u6216\u8f93\u5165\u6570\u636e\u3002<\/li><li><strong>\u6a21\u578b\u56fe\u8bb0\u5f55<\/strong>\uff1a\u80fd\u591f\u8bb0\u5f55\u548c\u53ef\u89c6\u5316\u6a21\u578b\u7684\u8ba1\u7b97\u56fe\uff0c\u8fd9\u5bf9\u4e8e\u7406\u89e3\u6a21\u578b\u7684\u7ed3\u6784\u548c\u6570\u636e\u6d41\u975e\u5e38\u6709\u5e2e\u52a9\u3002<\/li><li><strong>\u5206\u5e03\u8bb0\u5f55<\/strong>\uff1a\u53ef\u4ee5\u8bb0\u5f55\u6570\u636e\u7684\u5206\u5e03\u60c5\u51b5\uff0c\u5e2e\u52a9\u89c2\u5bdf\u6a21\u578b\u5728\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u53c2\u6570\u6216\u635f\u5931\u7684\u53d8\u5316\u3002<\/li><li><strong>\u76f4\u65b9\u56fe\u8bb0\u5f55<\/strong>\uff1a\u53ef\u4ee5\u8bb0\u5f55\u6743\u91cd\u3001\u68af\u5ea6\u7b49\u7684\u76f4\u65b9\u56fe\uff0c\u65b9\u4fbf\u89c2\u5bdf\u8fd9\u4e9b\u53c2\u6570\u5728\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u7684\u53d8\u5316\u3002<\/li><\/ol>\n\n\n\n<figure class=\"wp-block-image size-large\"><img decoding=\"async\" src=\"https:\/\/sanroimg.oss-cn-shanghai.aliyuncs.com\/img\/202410301343339.png\" alt=\"\"\/><\/figure>\n\n\n\n<p>\u793a\u4f8b\u4ee3\u7801<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code>import torch<br>from torch.utils.tensorboard import SummaryWriter<br>\u200b<br># \u521b\u5efa\u4e00\u4e2a SummaryWriter \u5b9e\u4f8b<br>writer = SummaryWriter('logs')<br>\u200b<br># \u8bb0\u5f55\u8bad\u7ec3\u635f\u5931<br>for epoch in range(10):<br> &nbsp; &nbsp;loss = 0.1 * (10 - epoch) &nbsp;# \u793a\u4f8b\u635f\u5931<br> &nbsp; &nbsp;writer.add_scalar('Loss\/train', loss, epoch)<br>\u200b<br># \u8bb0\u5f55\u6a21\u578b\u56fe<br>dummy_input = torch.randn(1, 3, 224, 224) &nbsp;# \u793a\u4f8b\u8f93\u5165<br>model = ... &nbsp;# \u5047\u8bbe model \u662f\u4f60\u7684\u6a21\u578b<br>writer.add_graph(model, dummy_input)<br>\u200b<br># \u5173\u95ed SummaryWriter<br>writer.close()<br>\u200b<\/code><\/pre>\n\n\n\n<p>\u547d\u4ee4\u884c\u8fd0\u884c<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code>tensorboard --logdir=logs<\/code><\/pre>\n","protected":false},"excerpt":{"rendered":"<p>torch.utils.tens&#46;&#46;&#46;<\/p>\n","protected":false},"author":1,"featured_media":2454,"comment_status":"closed","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[47,46,4],"tags":[],"class_list":["post-2451","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-cuda","category-pytorch","category-4"],"_links":{"self":[{"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/posts\/2451","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/comments?post=2451"}],"version-history":[{"count":0,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/posts\/2451\/revisions"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/"}],"wp:attachment":[{"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/media?parent=2451"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/categories?post=2451"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/sanlangcode.com\/index.php\/wp-json\/wp\/v2\/tags?post=2451"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}