keras 设置模型每层的trainable

最新推荐文章于 2023-03-19 11:39:28 发布

原创最新推荐文章于 2023-03-19 11:39:28 发布 · 1.4k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#keras

python3 专栏收录该内容

39 篇文章

订阅专栏

本文介绍如何使用Python管理深度学习模型的训练过程，包括将所有层设为不可训练，然后针对特定层设置可训练状态，并分别列出 trainable 和 untrainable 层的权重名称。适合理解模型训练控制的开发者。

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