param.requires_grad = False的作用是:
屏蔽預訓練模型的權重。
只訓練最後一層的全連接的權重。
最後一層的添加,是通過[2]中的一句代碼:
model_conv.fc = nn.Linear(num_ftrs, 2)
可以發現,凍結預訓練模型的權重後,驗證精度會下降。
param.requires_grad = False的作用是:
屏蔽預訓練模型的權重。
只訓練最後一層的全連接的權重。
最後一層的添加,是通過[2]中的一句代碼:
model_conv.fc = nn.Linear(num_ftrs, 2)
可以發現,凍結預訓練模型的權重後,驗證精度會下降。