keras 权重冻结

 

# model.layers可以获得模型的某一个层，你可以打印出层的名字，并且使用layer.trainable开关冻结，True为正常训练，False为冻结。
for layer in model2.layers:
    layer.trainable = False
    print layer.trainable
    print layer.name

或者在函数模型中直接关闭

x=Convolution2D(32,[3,3],strides=(1,1),
                          padding='same',
                          use_bias=False,
                          name='layer1')(inputs)
x.trainable=False

或者在卷积中关闭

x=Convolution2D(32,[3,3],strides=(1,1),
                          padding='same',
                          use_bias=False,
                          name='layer1'，trainable=False)(inputs)