输出 Keras 模型的训练参数

Question

如果我在 GRU 模型上进行训练，有没有一种方法可以输出学习到的参数，以便下次使用更多数据进行训练时，可以以这些学习到的参数为起点进行初始化？

Answer 1

1. 如果保存整个模型：

   weights = model.get_weights() # Getting params
   model.set_weights(weights) # Setting params
   

2. 如果保存单层：你需要找到要保存的层的索引（假设是i），然后：

   weights = model.layers[i].get_weights() # Getting params
   model.layers[i].set_weights(weights) # Setting params