在 keras 模型中,可以在编译时设置模型的学习率,如下所示,
model.compile(optimizer=Adam(learning_rate=0.001), loss=...)
Run Code Online (Sandbox Code Playgroud)
这为模型中的所有层设置了相同的学习率,但是如何为模型的每一层设置不同的学习率?
像这样,
第 1 层:0.001
第 2 层:0.05
第 3 层:0.02
等等。
我该如何做到这一点keras
?或在tf.keras
?