更改Keras中现有模型的辍学率的最简单方法?

时间:2019-04-18 21:23:06

标签: tensorflow keras

看来

model.layers[n].rate
可以查看和更改

,但它不能到达后端并实际上改变了训练行为。真正更改它的最简单方法是什么?我希望不必制作一个全新的模型并转移权重。

1 个答案:

答案 0 :(得分:2)

最简单的方法是:

  1. 更改图层的费率 model.layers[i].rate = 0.04 #layer[i] is the dropout layer

  2. 使用以下命令将此模型克隆为新模型 model = keras.models.clone(model) #weights would be reinitialized

  3. 编译新模型 model.compile(optimizer=..., loss=...) #optimizer state would be reset

  4. 将原始权重设置为新的克隆模型 model.load_weights(file_weights) #load weights

可以在here上找到有关此问题的讨论。