代碼來(lái)自[1],

[1]的詳細(xì)註釋在[2]中

?

param.requires_grad = False的作用是:

屏蔽預(yù)訓(xùn)練模型的權(quán)重

只訓(xùn)練最後一層的全連接層的權(quán)重.

?

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

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

可以發(fā)現(xiàn)，凍結(jié)預(yù)訓(xùn)練模型的權(quán)重後，驗(yàn)證精度會(huì)下降．

?

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

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。