{"id":700,"date":"2021-12-25T14:37:44","date_gmt":"2021-12-25T06:37:44","guid":{"rendered":"http:\/\/139.9.1.231\/?p=700"},"modified":"2021-12-25T14:37:45","modified_gmt":"2021-12-25T06:37:45","slug":"torchmodel","status":"publish","type":"post","link":"http:\/\/139.9.1.231\/index.php\/2021\/12\/25\/torchmodel\/","title":{"rendered":"torch\u6a21\u578b\u7684\u4fdd\u5b58\u548c\u8bfb\u53d6"},"content":{"rendered":"\n<p>pytorch\u4e2d\u6a21\u578b\u7684\u4fdd\u5b58\u548c\u8bfb\u53d6\uff1atorch.load    torch.save<\/p>\n\n\n\n<p>1\u3001\u8bfb\u53d6tensor<\/p>\n\n\n\n<p>      \u6211\u4eec\u53ef\u4ee5\u76f4\u63a5\u4f7f\u7528<code>save<\/code>\u51fd\u6570\u548c<code>load<\/code>\u51fd\u6570\u5206\u522b\u5b58\u50a8\u548c\u8bfb\u53d6<code>Tensor<\/code>\u3002<code>save<\/code>\u4f7f\u7528Python\u7684pickle\u5b9e\u7528\u7a0b\u5e8f\u5c06\u5bf9\u8c61\u8fdb\u884c\u5e8f\u5217\u5316\uff0c\u7136\u540e\u5c06\u5e8f\u5217\u5316\u7684\u5bf9\u8c61\u4fdd\u5b58\u5230disk\uff0c\u4f7f\u7528<code>save<\/code>\u53ef\u4ee5\u4fdd\u5b58\u5404\u79cd\u5bf9\u8c61,\u5305\u62ec\u6a21\u578b\u3001\u5f20\u91cf\u548c\u5b57\u5178\u7b49\u3002\u800c<code>load<\/code>\u4f7f\u7528pickle unpickle\u5de5\u5177\u5c06pickle\u7684\u5bf9\u8c61\u6587\u4ef6\u53cd\u5e8f\u5217\u5316\u4e3a\u5185\u5b58<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code>import torch\r\nfrom torch import nn\r\n\r\nx = torch.ones(3)\r\ntorch.save(x, 'x.pt')\nx2 = torch.load('x.pt')\\\n\n\u5b58\u50a8\u4e00\u4e2aTensor\u5217\u8868\u5e76\u8bfb\u56de\u5185\u5b58\u3002\n<\/code><\/pre>\n\n\n\n<pre class=\"wp-block-code\"><code>y = torch.zeros(4)\r\ntorch.save(&#91;x, y], 'xy.pt')\r\nxy_list = torch.load('xy.pt')\r\nxy_list\n\n\u5b58\u50a8\u5e76\u8bfb\u53d6\u4e00\u4e2a\u4ece\u5b57\u7b26\u4e32\u6620\u5c04\u5230Tensor\u7684\u5b57\u5178\u3002\n\ntorch.save({'x': x, 'y': y}, 'xy_dict.pt')\r\nxy = torch.load('xy_dict.pt')\r\nxy<\/code><\/pre>\n\n\n\n<p>\u8bfb\u5199\u6a21\u578b\uff1a<\/p>\n\n\n\n<p>\u5728PyTorch\u4e2d\uff0c<code>Module<\/code>\u7684\u53ef\u5b66\u4e60\u53c2\u6570(\u5373\u6743\u91cd\u548c\u504f\u5dee)\uff0c\u6a21\u5757\u6a21\u578b\u5305\u542b\u5728\u53c2\u6570\u4e2d(\u901a\u8fc7<code>model.parameters()<\/code>\u8bbf\u95ee)\u3002<code>state_dict<\/code>\u662f\u4e00\u4e2a\u4ece\u53c2\u6570\u540d\u79f0\u9690\u5c04\u5230\u53c2\u6570<code>Tesnor<\/code>\u7684\u5b57\u5178\u5bf9\u8c61\u3002<\/p>\n\n\n\n<p>1\u3001\u5c06\u6a21\u578b\u548c\u53c2\u6570\u90fd\u4fdd\u5b58\u548c\u8bfb\u53d6<\/p>\n\n\n\n<p>torch.save(model, PATH)<\/p>\n\n\n\n<p>model = torch.load(PATH)<\/p>\n\n\n\n<p>2\u3001\u53ea\u5b58\u50a8\u6a21\u578b\u53c2\u6570(<code>state_dict<\/code>)<\/p>\n\n\n\n<p>torch.save(model.state_dict(), PATH) # \u63a8\u8350\u7684\u6587\u4ef6\u540e\u7f00\u540d\u662fpt\u6216pth<\/p>\n\n\n\n<p>\u52a0\u8f7d\uff1a<\/p>\n\n\n\n<p>model = TheModelClass(*args, **kwargs)<\/p>\n\n\n\n<p> model.load_state_dict(torch.load(PATH))<\/p>\n\n\n\n<div class=\"wp-block-image is-style-default\"><figure class=\"aligncenter size-large\"><img src=\"https:\/\/cdn.pixabay.com\/photo\/2021\/10\/26\/12\/34\/christmas-6743572_1280.jpg\" alt=\"\"\/><figcaption>\u5723\u8bde\u5feb\u4e50<\/figcaption><\/figure><\/div>\n\n\n\n<p><\/p>\n","protected":false},"excerpt":{"rendered":"<p>pytorch\u4e2d\u6a21\u578b\u7684\u4fdd\u5b58\u548c\u8bfb\u53d6\uff1atorch.load torch.save 1\u3001\u8bfb\u53d6tensor \u6211\u4eec\u53ef\u4ee5\u76f4 &hellip; <a href=\"http:\/\/139.9.1.231\/index.php\/2021\/12\/25\/torchmodel\/\" class=\"more-link\">\u7ee7\u7eed\u9605\u8bfb<span class=\"screen-reader-text\">torch\u6a21\u578b\u7684\u4fdd\u5b58\u548c\u8bfb\u53d6<\/span><\/a><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":[],"categories":[11],"tags":[],"_links":{"self":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/700"}],"collection":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/comments?post=700"}],"version-history":[{"count":18,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/700\/revisions"}],"predecessor-version":[{"id":718,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/700\/revisions\/718"}],"wp:attachment":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/media?parent=700"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/categories?post=700"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/tags?post=700"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}