{"id":5284,"date":"2024-03-14T02:37:19","date_gmt":"2024-03-14T02:37:19","guid":{"rendered":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/"},"modified":"2025-08-01T13:04:52","modified_gmt":"2025-08-01T13:04:52","slug":"how-to-implement-transfer-learning-in-pytorch","status":"publish","type":"post","link":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/","title":{"rendered":"PyTorch Transfer Learning: Step-by-Step Guide"},"content":{"rendered":"

In PyTorch, implementing transfer learning can typically be accomplished through the following steps:<\/p>\n

    \n
  1. Load a pre-trained model: Begin by loading a model that has been pre-trained on a large-scale data set, such as ResNet or VGG trained on ImageNet.<\/li>\n
  2. Modify the model architecture: Adjust the last few layers of the pre-trained model to meet the output requirements of the new task based on the specific task at hand.<\/li>\n
  3. Freeze the model weights: Lock the weights of the pre-trained model so that they will not be updated during the training process.<\/li>\n
  4. Define a new loss function based on the requirements of the new task.<\/li>\n
  5. Training the model: Train the modified model using a new dataset, updating only the weights of the newly added layers.<\/li>\n
  6. Fine-tune the model: To further improve the model’s performance, you can unfreeze some of the pre-trained model weights and continue training the entire model.<\/li>\n<\/ol>\n

    Here is a simple example code to demonstrate how to implement transfer learning in PyTorch.<\/p>\n

    import<\/span> torch\r\nimport<\/span> torch.nn as<\/span> nn\r\nimport<\/span> torchvision.models as<\/span> models\r\nimport<\/span> torchvision.transforms as<\/span> transforms\r\nimport<\/span> torch.optim as<\/span> optim\r\nimport<\/span> torch.utils.data as<\/span> data\r\nfrom<\/span> torchvision.datasets import<\/span> ImageFolder\r\n\r\n# \u52a0\u8f7d\u9884\u8bad\u7ec3\u6a21\u578b<\/span>\r\npretrained_model = models.resnet18(pretrained=True<\/span>)\r\n\r\n# \u4fee\u6539\u6a21\u578b\u7ed3\u6784<\/span>\r\nnum_ftrs = pretrained_model.fc.in_features\r\npretrained_model.fc = nn.Linear(num_ftrs, 2<\/span>)  # \u5047\u8bbe\u65b0\u4efb\u52a1\u662f\u4e00\u4e2a\u4e8c\u5206\u7c7b\u95ee\u9898<\/span>\r\n\r\n# \u51bb\u7ed3\u6a21\u578b\u6743\u91cd<\/span>\r\nfor<\/span> param in<\/span> pretrained_model.parameters():\r\n    param.requires_grad = False<\/span>\r\n\r\n# \u52a0\u8f7d\u6570\u636e<\/span>\r\ntransform = transforms.Compose([\r\n    transforms.Resize(256<\/span>),\r\n    transforms.CenterCrop(224<\/span>),\r\n    transforms.ToTensor()\r\n])\r\ntrain_dataset = ImageFolder('path_to_train_data'<\/span>, transform=transform)\r\ntrain_loader = data.DataLoader(train_dataset, batch_size=32<\/span>, shuffle=True<\/span>)\r\n\r\n# \u5b9a\u4e49\u635f\u5931\u51fd\u6570\u548c\u4f18\u5316\u5668<\/span>\r\ncriterion = nn.CrossEntropyLoss()\r\noptimizer = optim.SGD(pretrained_model.fc.parameters(), lr=0.001<\/span>)\r\n\r\n# \u8bad\u7ec3\u6a21\u578b<\/span>\r\npretrained_model.train()\r\nfor<\/span> epoch in<\/span> range<\/span>(10<\/span>):\r\n    for<\/span> inputs, labels in<\/span> train_loader:\r\n        optimizer.zero_grad()\r\n        outputs = pretrained_model(inputs)\r\n        loss = criterion(outputs, labels)\r\n        loss.backward()\r\n        optimizer.step()\r\n\r\n# \u4fdd\u5b58\u6a21\u578b<\/span>\r\ntorch.save(pretrained_model.state_dict(), 'pretrained_model.pth'<\/span>)\r\n<\/code><\/pre>\n
    This is a simple example of transfer learning that can be adjusted and optimized based on specific circumstances in practice.<\/p>\n","protected":false},"excerpt":{"rendered":"
    In PyTorch, implementing transfer learning can typically be accomplished through the following steps: Load a pre-trained model: Begin by loading a model that has been pre-trained on a large-scale data set, such as ResNet or VGG trained on ImageNet. Modify the model architecture: Adjust the last few layers of the pre-trained model to meet the […]<\/p>\n","protected":false},"author":5,"featured_media":0,"comment_status":"closed","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"_import_markdown_pro_load_document_selector":0,"_import_markdown_pro_submit_text_textarea":"","footnotes":""},"categories":[1],"tags":[960,75,1218,1239,1259],"class_list":["post-5284","post","type-post","status-publish","format-standard","hentry","category-uncategorized","tag-deep-learning","tag-machine-learning","tag-pre-trained-models","tag-pytorch","tag-transfer-learning"],"yoast_head":"\nPyTorch Transfer Learning: Step-by-Step Guide - Blog - Silicon Cloud<\/title>\n<meta name=\"description\" content=\"Master PyTorch transfer learning: Load pre-trained models, modify layers, freeze weights & implement efficiently. Tutorial included.\" \/>\n<meta name=\"robots\" content=\"index, follow, max-snippet:-1, max-image-preview:large, max-video-preview:-1\" \/>\n<link rel=\"canonical\" href=\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/\" \/>\n<meta property=\"og:locale\" content=\"en_US\" \/>\n<meta property=\"og:type\" content=\"article\" \/>\n<meta property=\"og:title\" content=\"PyTorch Transfer Learning: Step-by-Step Guide\" \/>\n<meta property=\"og:description\" content=\"Master PyTorch transfer learning: Load pre-trained models, modify layers, freeze weights & implement efficiently. Tutorial included.\" \/>\n<meta property=\"og:url\" content=\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/\" \/>\n<meta property=\"og:site_name\" content=\"Blog - Silicon Cloud\" \/>\n<meta property=\"article:publisher\" content=\"https:\/\/www.facebook.com\/SiliCloudGlobal\/\" \/>\n<meta property=\"article:published_time\" content=\"2024-03-14T02:37:19+00:00\" \/>\n<meta property=\"article:modified_time\" content=\"2025-08-01T13:04:52+00:00\" \/>\n<meta name=\"author\" content=\"Emily Johnson\" \/>\n<meta name=\"twitter:card\" content=\"summary_large_image\" \/>\n<meta name=\"twitter:creator\" content=\"@SiliCloudGlobal\" \/>\n<meta name=\"twitter:site\" content=\"@SiliCloudGlobal\" \/>\n<meta name=\"twitter:label1\" content=\"Written by\" \/>\n\t<meta name=\"twitter:data1\" content=\"Emily Johnson\" \/>\n\t<meta name=\"twitter:label2\" content=\"Est. reading time\" \/>\n\t<meta name=\"twitter:data2\" content=\"1 minute\" \/>\n<script type=\"application\/ld+json\" class=\"yoast-schema-graph\">{\"@context\":\"https:\/\/schema.org\",\"@graph\":[{\"@type\":\"Article\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/#article\",\"isPartOf\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/\"},\"author\":{\"name\":\"Emily Johnson\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/3b041b19cffc258705478ecfab895378\"},\"headline\":\"PyTorch Transfer Learning: Step-by-Step Guide\",\"datePublished\":\"2024-03-14T02:37:19+00:00\",\"dateModified\":\"2025-08-01T13:04:52+00:00\",\"mainEntityOfPage\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/\"},\"wordCount\":188,\"publisher\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/#organization\"},\"keywords\":[\"Deep Learning\",\"machine learning\",\"pre-trained models\",\"PyTorch\",\"transfer learning\"],\"inLanguage\":\"en-US\"},{\"@type\":\"WebPage\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/\",\"url\":\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/\",\"name\":\"PyTorch Transfer Learning: Step-by-Step Guide - Blog - Silicon Cloud\",\"isPartOf\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/#website\"},\"datePublished\":\"2024-03-14T02:37:19+00:00\",\"dateModified\":\"2025-08-01T13:04:52+00:00\",\"description\":\"Master PyTorch transfer learning: Load pre-trained models, modify layers, freeze weights & implement efficiently. Tutorial included.\",\"breadcrumb\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/#breadcrumb\"},\"inLanguage\":\"en-US\",\"potentialAction\":[{\"@type\":\"ReadAction\",\"target\":[\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/\"]}]},{\"@type\":\"BreadcrumbList\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/#breadcrumb\",\"itemListElement\":[{\"@type\":\"ListItem\",\"position\":1,\"name\":\"Home\",\"item\":\"https:\/\/www.silicloud.com\/blog\/\"},{\"@type\":\"ListItem\",\"position\":2,\"name\":\"PyTorch Transfer Learning: Step-by-Step Guide\"}]},{\"@type\":\"WebSite\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#website\",\"url\":\"https:\/\/www.silicloud.com\/blog\/\",\"name\":\"Silicon Cloud Blog\",\"description\":\"\",\"publisher\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/#organization\"},\"inLanguage\":\"en-US\"},{\"@type\":\"Organization\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#organization\",\"name\":\"Silicon Cloud Blog\",\"url\":\"https:\/\/www.silicloud.com\/blog\/\",\"logo\":{\"@type\":\"ImageObject\",\"inLanguage\":\"en-US\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/logo\/image\/\",\"url\":\"https:\/\/www.silicloud.com\/blog\/wp-content\/uploads\/2023\/11\/EN-SILICON-Full.png\",\"contentUrl\":\"https:\/\/www.silicloud.com\/blog\/wp-content\/uploads\/2023\/11\/EN-SILICON-Full.png\",\"width\":1024,\"height\":1024,\"caption\":\"Silicon Cloud Blog\"},\"image\":{\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/logo\/image\/\"},\"sameAs\":[\"https:\/\/www.facebook.com\/SiliCloudGlobal\/\",\"https:\/\/twitter.com\/SiliCloudGlobal\"]},{\"@type\":\"Person\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/3b041b19cffc258705478ecfab895378\",\"name\":\"Emily Johnson\",\"image\":{\"@type\":\"ImageObject\",\"inLanguage\":\"en-US\",\"@id\":\"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/image\/\",\"url\":\"https:\/\/secure.gravatar.com\/avatar\/a5cb4e73d02ab1d79f2dfe919389ff7c1de072baa97686392031c03d858cc358?s=96&d=mm&r=g\",\"contentUrl\":\"https:\/\/secure.gravatar.com\/avatar\/a5cb4e73d02ab1d79f2dfe919389ff7c1de072baa97686392031c03d858cc358?s=96&d=mm&r=g\",\"caption\":\"Emily Johnson\"},\"url\":\"https:\/\/www.silicloud.com\/blog\/author\/emilyjohnson\/\"}]}<\/script>\n<!-- \/ Yoast SEO Premium plugin. -->","yoast_head_json":{"title":"PyTorch Transfer Learning: Step-by-Step Guide - Blog - Silicon Cloud","description":"Master PyTorch transfer learning: Load pre-trained models, modify layers, freeze weights & implement efficiently. Tutorial included.","robots":{"index":"index","follow":"follow","max-snippet":"max-snippet:-1","max-image-preview":"max-image-preview:large","max-video-preview":"max-video-preview:-1"},"canonical":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/","og_locale":"en_US","og_type":"article","og_title":"PyTorch Transfer Learning: Step-by-Step Guide","og_description":"Master PyTorch transfer learning: Load pre-trained models, modify layers, freeze weights & implement efficiently. Tutorial included.","og_url":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/","og_site_name":"Blog - Silicon Cloud","article_publisher":"https:\/\/www.facebook.com\/SiliCloudGlobal\/","article_published_time":"2024-03-14T02:37:19+00:00","article_modified_time":"2025-08-01T13:04:52+00:00","author":"Emily Johnson","twitter_card":"summary_large_image","twitter_creator":"@SiliCloudGlobal","twitter_site":"@SiliCloudGlobal","twitter_misc":{"Written by":"Emily Johnson","Est. reading time":"1 minute"},"schema":{"@context":"https:\/\/schema.org","@graph":[{"@type":"Article","@id":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/#article","isPartOf":{"@id":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/"},"author":{"name":"Emily Johnson","@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/3b041b19cffc258705478ecfab895378"},"headline":"PyTorch Transfer Learning: Step-by-Step Guide","datePublished":"2024-03-14T02:37:19+00:00","dateModified":"2025-08-01T13:04:52+00:00","mainEntityOfPage":{"@id":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/"},"wordCount":188,"publisher":{"@id":"https:\/\/www.silicloud.com\/blog\/#organization"},"keywords":["Deep Learning","machine learning","pre-trained models","PyTorch","transfer learning"],"inLanguage":"en-US"},{"@type":"WebPage","@id":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/","url":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/","name":"PyTorch Transfer Learning: Step-by-Step Guide - Blog - Silicon Cloud","isPartOf":{"@id":"https:\/\/www.silicloud.com\/blog\/#website"},"datePublished":"2024-03-14T02:37:19+00:00","dateModified":"2025-08-01T13:04:52+00:00","description":"Master PyTorch transfer learning: Load pre-trained models, modify layers, freeze weights & implement efficiently. Tutorial included.","breadcrumb":{"@id":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/#breadcrumb"},"inLanguage":"en-US","potentialAction":[{"@type":"ReadAction","target":["https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/"]}]},{"@type":"BreadcrumbList","@id":"https:\/\/www.silicloud.com\/blog\/how-to-implement-transfer-learning-in-pytorch\/#breadcrumb","itemListElement":[{"@type":"ListItem","position":1,"name":"Home","item":"https:\/\/www.silicloud.com\/blog\/"},{"@type":"ListItem","position":2,"name":"PyTorch Transfer Learning: Step-by-Step Guide"}]},{"@type":"WebSite","@id":"https:\/\/www.silicloud.com\/blog\/#website","url":"https:\/\/www.silicloud.com\/blog\/","name":"Silicon Cloud Blog","description":"","publisher":{"@id":"https:\/\/www.silicloud.com\/blog\/#organization"},"inLanguage":"en-US"},{"@type":"Organization","@id":"https:\/\/www.silicloud.com\/blog\/#organization","name":"Silicon Cloud Blog","url":"https:\/\/www.silicloud.com\/blog\/","logo":{"@type":"ImageObject","inLanguage":"en-US","@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/logo\/image\/","url":"https:\/\/www.silicloud.com\/blog\/wp-content\/uploads\/2023\/11\/EN-SILICON-Full.png","contentUrl":"https:\/\/www.silicloud.com\/blog\/wp-content\/uploads\/2023\/11\/EN-SILICON-Full.png","width":1024,"height":1024,"caption":"Silicon Cloud Blog"},"image":{"@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/logo\/image\/"},"sameAs":["https:\/\/www.facebook.com\/SiliCloudGlobal\/","https:\/\/twitter.com\/SiliCloudGlobal"]},{"@type":"Person","@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/3b041b19cffc258705478ecfab895378","name":"Emily Johnson","image":{"@type":"ImageObject","inLanguage":"en-US","@id":"https:\/\/www.silicloud.com\/blog\/#\/schema\/person\/image\/","url":"https:\/\/secure.gravatar.com\/avatar\/a5cb4e73d02ab1d79f2dfe919389ff7c1de072baa97686392031c03d858cc358?s=96&d=mm&r=g","contentUrl":"https:\/\/secure.gravatar.com\/avatar\/a5cb4e73d02ab1d79f2dfe919389ff7c1de072baa97686392031c03d858cc358?s=96&d=mm&r=g","caption":"Emily Johnson"},"url":"https:\/\/www.silicloud.com\/blog\/author\/emilyjohnson\/"}]}},"_links":{"self":[{"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/posts\/5284","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/users\/5"}],"replies":[{"embeddable":true,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/comments?post=5284"}],"version-history":[{"count":2,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/posts\/5284\/revisions"}],"predecessor-version":[{"id":150026,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/posts\/5284\/revisions\/150026"}],"wp:attachment":[{"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/media?parent=5284"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/categories?post=5284"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.silicloud.com\/blog\/wp-json\/wp\/v2\/tags?post=5284"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}