How to save momentum during training in Keras?
Question
I want to be able to save the state of training my neural net using SGD. I know I can save the weights using the save_weights function. What I don't know is how to save the momentum of the gradient decent. Does anyone know how to do this?
1 answers
solution1
1 2016-05-11 04:59:11
使用model.optimizer.get_state()和model.optimizer.set_state() - 请参阅https://github.com/fchollet/keras/issues/454 。
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.
Related Question
How to keep a list of models saved during training (save best only) in Keras?
How to record the values of a tensor during training in Keras?
How to save training history on every epoch in Keras?
Keras: How to save model and continue training?
How to replace loss function during training tensorflow.keras
How to randomly set inputs to zero in keras during training autoencoder (callback)?
How to call reset_states() during training in keras?
How do I print inside the loss function during training in Keras?
How to prevent Keras from computing metrics during training
How to get progress of current epoch during tf.keras training?
Related Tags