transfer learning pytorch
Author:

Transfer Learning for Computer Vision Tutorial ============================================== **Author**: `Sasank Chilamkurthy`_ In this tutorial, you will learn how to train a convolutional neural network for image classification using transfer learning. You can read more about the transfer learning at `cs231n notes <https://cs231n.github.io/transfer-learning/>`__ Quoting these notes, In practice, very few people train an entire Convolutional Network from scratch (with random initialization), because it is relatively rare to have a dataset of sufficient size. Instead, it is common to pretrain a ConvNet on a very large dataset (e.g. ImageNet, which contains 1.2 million images with 1000 categories), and then use the ConvNet either as an initialization or a fixed feature extractor for the task of interest. These two major transfer learning scenarios look as follows: - **Finetuning the ConvNet**: Instead of random initialization, we initialize the network with a pretrained network, like the one that is trained on imagenet 1000 dataset. Rest of the training looks as usual. - **ConvNet as fixed feature extractor**: Here, we will freeze the weights for all of the network except that of the final fully connected layer. This last fully connected layer is replaced with a new one with random weights and only this layer is trained.
Tasks: Deep Learning Fundamentals, Transfer Learning, Image Classification
Task Categories: Deep Learning Fundamentals, Computer Vision
Published: 10/07/23
Tags
pytorch
intermediate
Loading...