Update transfer_learning_tutorial.py

Author: kli-casia

beginner_source/transfer_learning_tutorial.py

@@ -297,7 +297,7 @@ def optim_scheduler_ft(model, epoch, init_lr=0.001, lr_decay_epoch=7):
     param.requires_grad = False
 
 # Parameters of newly constructed modules have requires_grad=True by default
-model.fc = nn.Linear(512, 100)
+model.fc = nn.Linear(512, 2)
 
 if use_gpu:
     model = model.cuda()