我试图用Keras和mnist数据集冻结预测模型中某个图层的权重,但它不起作用。代码如下:来自keras.layers导入Dense,Flatten来自keras ….
您需要在设置'trainable'后编译图形。 https://keras.io/getting-started/faq/#how-can-i-freeze-keras-layers
你可以使用:
model=Sequential() layer=Dense(64,init='glorot_uniform',input_shape=(784,)) layer.trainable=False model.add(layer) layer2=Dense(784, activation='sigmoid',init='glorot_uniform') layer2.trainable=True model.add(layer2) model.compile(loss='relu', optimizer=sgd,metrics = ['mae'])