自定义 KERAS 模型的可训练参数

时间:2021-02-05 02:04:03

标签: tensorflow keras neural-network tf.keras transfer-learning

Multi hidden units

我只想将所有参数设置为不可训练,除了连接到dense_5层的参数,它们是它自己的参数,以及连接到dense_5的dense_6层的参数。如何将这些最后的参数设置为 trainable ?通过将dense_6层的trainable设置为true,所有参数即使没有连接到dense_5也变得可训练。

1 个答案:

答案 0 :(得分:0)

您不能更改部分图层的 trainable。改变架构:不要连接dense_5,在dense_5之后添加dense_7,连接dense_6和dense_7的输出。然后为dense_5 和Dense_7 设置trainable

相关问题