keras 设置模型每层的trainable

def setTrainable(model,trainlayers):
    # 先全部设置成不可训练
    for layer in model.layers:
        layer.trainable = False
    # 设置可训练的层
    for i in trainlayers:
        model.layers[i].trainable = True

    # 可训练层
    print('trainable:')
    for x in model.trainable_weights:
        print(x.name)
    print('\n')

    # 不可训练层
    print('untrainable: ')
    for x in model.non_trainable_weights:
        print(x.name)
    print('\n')