且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

如何使用Keras冻结特定图层中的权重?

更新时间:2023-12-01 21:31:58

您可以使用:

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'])