{"id":628,"date":"2023-03-03T18:07:00","date_gmt":"2023-03-03T10:07:00","guid":{"rendered":"https:\/\/scutvk.cn\/?p=628"},"modified":"2023-03-15T11:05:01","modified_gmt":"2023-03-15T03:05:01","slug":"pytorch%e6%80%8e%e4%b9%88%e4%bd%bf%e7%94%a8%e8%87%aa%e5%8a%a8%e6%b7%b7%e5%90%88%e7%b2%be%e5%ba%a6","status":"publish","type":"post","link":"https:\/\/scutvk.cn\/?p=628","title":{"rendered":"Pytorch\u600e\u4e48\u4f7f\u7528\u81ea\u52a8\u6df7\u5408\u7cbe\u5ea6"},"content":{"rendered":"\n<p>PyTorch\u4e2d\u7684\u81ea\u52a8\u7cbe\u5ea6\u529f\u80fd\u662f\u901a\u8fc7\u6df7\u5408\u7cbe\u5ea6\u8bad\u7ec3\uff08Mixed Precision Training\uff09\u5b9e\u73b0\u7684\uff0c\u53ef\u4ee5\u4f7f\u7528\u4ee5\u4e0b\u6b65\u9aa4\u542f\u7528\u5b83\uff1a<\/p>\n\n\n\n<ol>\n<li>\u5bfc\u5165\u5fc5\u8981\u7684\u5e93\uff1a<\/li>\n<\/ol>\n\n\n\n<pre class=\"wp-block-code\"><code lang=\"python\" class=\"language-python\">import torch\nfrom torch.cuda.amp import autocast, GradScaler\n<\/code><\/pre>\n\n\n\n<ol start=\"2\">\n<li>\u521d\u59cb\u5316\u6df7\u5408\u7cbe\u5ea6\u8bbe\u7f6e\uff1a<\/li>\n<\/ol>\n\n\n\n<pre class=\"wp-block-code\"><code lang=\"python\" class=\"language-python\">scaler = GradScaler()<\/code><\/pre>\n\n\n\n<ol start=\"3\">\n<li>\u5728\u8bad\u7ec3\u5faa\u73af\u4e2d\u4f7f\u7528autocast\u548cGradScaler\uff1a<\/li>\n<\/ol>\n\n\n\n<pre class=\"wp-block-code\"><code lang=\"python\" class=\"language-python\">for data, target in data_loader:\n    # \u5c06\u6570\u636e\u548c\u76ee\u6807\u79fb\u52a8\u5230GPU\u4e0a\n    data, target = data.to(device), target.to(device)\n\n    # \u6a21\u578b\u63a8\u7406\n    with autocast():\n        output = model(data)\n        loss = criterion(output, target)\n\n    # \u53cd\u5411\u4f20\u64ad\u548c\u6743\u91cd\u66f4\u65b0\n    scaler.scale(loss).backward()\n    scaler.step(optimizer)\n    scaler.update()\n\n    # \u6253\u5370\u635f\u5931\n    print('Loss: {:.4f}'.format(loss.item()))\n<\/code><\/pre>\n\n\n\n<p>\u5728\u8fd9\u91cc\uff0c<code>with autocast()<\/code> \u5757\u5185\u7684\u64cd\u4f5c\u5c06\u5728\u81ea\u52a8\u6df7\u5408\u7cbe\u5ea6\u4e0b\u6267\u884c\uff0c\u800c <code>scaler.scale()<\/code> \u548c <code>scaler.step()<\/code> \u65b9\u6cd5\u5c06\u4f7f\u7528\u6b63\u786e\u7684\u7f29\u653e\u56e0\u5b50\u5bf9\u635f\u5931\u8fdb\u884c\u7f29\u653e\u5e76\u66f4\u65b0\u6a21\u578b\u7684\u6743\u91cd\u3002\u6700\u540e\uff0c<code>scaler.update()<\/code> \u65b9\u6cd5\u5c06\u66f4\u65b0\u7f29\u653e\u5668\u4ee5\u786e\u4fdd\u6b63\u786e\u7684\u7f29\u653e\u56e0\u5b50\u88ab\u4f7f\u7528\u3002<\/p>\n\n\n\n<p>\u4f7f\u7528\u81ea\u52a8\u7cbe\u5ea6\u53ef\u4ee5\u63d0\u9ad8\u6a21\u578b\u8bad\u7ec3\u7684\u901f\u5ea6\u548c\u6548\u7387\uff0c\u7279\u522b\u662f\u5728\u5904\u7406\u5927\u578b\u6a21\u578b\u548c\u6570\u636e\u96c6\u65f6\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>PyTorch\u4e2d\u7684\u81ea\u52a8\u7cbe\u5ea6\u529f\u80fd\u662f\u901a\u8fc7\u6df7\u5408\u7cbe\u5ea6\u8bad\u7ec3\uff08&hellip;<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"jetpack_publicize_message":"","jetpack_is_tweetstorm":false,"jetpack_publicize_feature_enabled":true,"jetpack_social_post_already_shared":true,"jetpack_social_options":{"image_generator_settings":{"template":"highway","enabled":false}}},"categories":[2,10,12],"tags":[],"jetpack_publicize_connections":[],"jetpack_featured_media_url":"","_links":{"self":[{"href":"https:\/\/scutvk.cn\/index.php?rest_route=\/wp\/v2\/posts\/628"}],"collection":[{"href":"https:\/\/scutvk.cn\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/scutvk.cn\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/scutvk.cn\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/scutvk.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=628"}],"version-history":[{"count":2,"href":"https:\/\/scutvk.cn\/index.php?rest_route=\/wp\/v2\/posts\/628\/revisions"}],"predecessor-version":[{"id":634,"href":"https:\/\/scutvk.cn\/index.php?rest_route=\/wp\/v2\/posts\/628\/revisions\/634"}],"wp:attachment":[{"href":"https:\/\/scutvk.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=628"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/scutvk.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=628"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/scutvk.cn\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=628"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}