除了我添加的图层外,如何固定VGG16相关模型的重量

时间:2019-04-10 16:02:57

标签: python keras conv-neural-network

除了最后一个Dense层以外,谁能告诉我如何固定模型的权重。

我的代码:

VGG_model=keras.applications.vgg16.VGG16(input_shape=(224,224,3),weights='imagenet', include_top=True)# Generate a model with all layers (with top)
#Add a layer where input is the output of the  second last layer 
x = Dense(2, activation='softmax', name='predictions')(VGG_model.layers[-2].output)

0 个答案:

没有答案