I used the example code mentioned in the tutorial except the model I changed to vgg16
Here is the link
https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html 21
I changed the line
model_ft = models.vgg16(pretrained=True)
->num_ftrs = model_ft.classifier[0].out_features
model_ft.fc = nn.Linear(num_ftrs, 2)
model_ft = model_ft.to(device)
Do I need to change any other lines to run VGG16 model ?
Do I need to change anoth