2017-04-18 47 views
1

當我嘗試實現轉移學習來訓練我的ConvNet模型時,我在這個Keras博客(goo.gl/N60H13)中沒有理解代碼片段。總的來說,VGG16有21層,但爲了凍結除最終FC之外的所有其他層,我需要凍結的層數爲20.有沒有人明白爲什麼博客使它成爲24([ 25])?VGG16轉移學習

代碼:

for layer in model.layers[:25]: 
    layer.trainable = False 

回答

0

也許有事情做與螻的框架。

從這gist,我們知道VGG16 model for Keras has been obtained by directly converting the Caffe model.所以,它最初也計數爲zero-padding layers。雙重檢查revision history,它確實包含那些zero-padding layers

對於從解冰的當前版本的VGG16,也許應該是

for layer in model.layers[:15]: 
    layer.trainable = False