Merge pull request #653 from jlin27/master

Add comment to clarify output_features for nn.Linear method

Author: brianjo

beginner_source/transfer_learning_tutorial.py

@@ -252,6 +252,8 @@ def visualize_model(model, num_images=6):
 
 model_ft = models.resnet18(pretrained=True)
 num_ftrs = model_ft.fc.in_features
+# Here the size of each output sample is set to 2.
+# Alternatively, it can be generalized to nn.Linear(num_ftrs, len(class_names)).
 model_ft.fc = nn.Linear(num_ftrs, 2)
 
 model_ft = model_ft.to(device)