6.3: fix issue about torchvision

source/第六章/6.3 模型微调-torchvision.md

@@ -112,7 +112,7 @@ model = models.resnet18(pretrained=True)
 set_parameter_requires_grad(model, feature_extract)
 # 修改模型
 num_ftrs = model.fc.in_features
-model.fc = nn.Linear(in_features=512, out_features=4, bias=True)
+model.fc = nn.Linear(in_features=num_ftrs, out_features=4, bias=True)
 ```
 
 之后在训练过程中，model仍会进行梯度回传，但是参数更新则只会发生在fc层。通过设定参数的requires_grad属性，我们完成了指定训练模型的特定层的目标，这对实现模型微调非常重要。