如何在使用VGG(Pytorch)的pretrain權重參數的同時變更網絡結構

pytorch vgg batchnormalization attention

acxyle 2020-01-08 17:57:45 ‧ 1028 瀏覽

如題。

疑惑源自 BatchNormalize 層的添加，在 torchvision.models 的 vgg.py 中，如果使用vgg16_bn，則會導致每一個 Convolution 層后都添加 bn 層，但是實際上我只想在網絡中的某幾個捲積層后添加。

一開始嘗試通過自己來寫VGG解決這個問題，但是在load_state_dict的時候又有問題，自己寫的網絡與 pretrain 參數的 frame 有差異，所以無法 load。

所以求助，如何在可以導入pretrain 權重參數的情況下，在 VGG 網絡結構中進行修改，如添加 bn, attention 等結構？

增廣建文 iT邦研究生 5 級 ‧ 2021-10-15 13:21:57 檢舉

fintune model的話只要用部分layer的weight就好

增廣建文 iT邦研究生 5 級 ‧ 2021-10-15 13:22:02 檢舉

https://discuss.pytorch.org/t/how-to-load-part-of-pre-trained-model/1113/2

直播研討會

{{ item.channelVendor }} {{ item.webinarstarted }} |

直播中

立即登入回答

參賽組數

1064 組

團體組數

40 組

累計文章數

22200 篇

完賽人數

602 人

IT邦幫忙