Deep Learning, Pytorch, Fine Tuning, Computer Vision, Transfer Learning.
new task num_classes = 10 model.fc = nn.Linear(model.fc.in_features, num_classes) Freezing only specific layers in the network : # Freeze specific layers (e.g.,the first two convolutional layers) of the pre-trained model for name, param in model.named_parameters(): if 'conv1' in name or 'layer1' in