param.requires_grad = False的作用

param.requires_grad = False的作用是:

屏蔽預訓練模型的權重。

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

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

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

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章