输出Keras模型的训练参数

时间:2018-02-01 14:53:06

标签: tensorflow keras

如果我正在训练GRU模型,有没有办法可以输出学习的参数,这样当我下次训练更多的数据时,我可以用这些学习的参数初始化作为起点?

1 个答案:

答案 0 :(得分: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