简体   繁体   English

如何在 Keras 训练期间保存动力?

[英]How to save momentum during training in Keras?

I want to be able to save the state of training my neural net using SGD.我希望能够使用 SGD 保存训练我的神经网络的状态。 I know I can save the weights using the save_weights function.我知道我可以使用 save_weights 函数保存权重。 What I don't know is how to save the momentum of the gradient decent.我不知道的是如何保存梯度体面的动量。 Does anyone know how to do this?有谁知道如何做到这一点?

使用model.optimizer.get_state()model.optimizer.set_state() - 请参阅https://github.com/fchollet/keras/issues/454

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM