如果我正在训练GRU模型,有没有办法可以输出学习的参数,这样当我下次训练更多的数据时,我可以用这些学习的参数初始化作为起点?
答案 0 :(得分:1)
如果保存整个模型:
weights = model.get_weights() # Getting params
model.set_weights(weights) # Setting params
如果保存单个图层:您需要找到要保存的图层的索引(让我们说它是i
),然后:
weights = model.layers[i].get_weights() # Getting params
model.layers[i].set_weights(weights) # Setting params