PyTorch遷移學習

如下是兩個主要的遷移學習場景： Finetuning the convnet: 我們使用預訓練網絡初始化網絡，而不是隨機初始化，就像在imagenet 1000數據集上訓練的網絡一樣。其餘訓練看起來像往常一樣。 ConvNet as fixed feature extractor: 在這裏，我們將凍結除最終完全連接層之外的所有網絡的權重。最後一個全連接層被替換爲具有隨機權重的新層，並且僅訓練該層。

>>阅读原文<<