代碼來自[1],

[1]的詳細註釋在[2]中

?

param.requires_grad = False的作用是:

屏蔽預訓練模型的權重

只訓練最後一層的全連接層的權重.

?

最後一層的添加，是通過[2]中的一句代碼:

model_conv.fc = nn.Linear(num_ftrs, 2)

可以發現，凍結預訓練模型的權重後，驗證精度會下降．

?

Reference:

[1]https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html

[2]https://github.com/appleyuchi/TPU_GPU_Code/blob/master/torchvision%E7%9A%84%E7%94%A8%E6%B3%95%E5%AF%A6%E4%BE%8B.ipynb

總結

以上是生活随笔為你收集整理的param.requires_grad = False的作用的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。