[英]How to keep a list of models saved during training (save best only) in Keras?
[英]How to save momentum during training in Keras?
我希望能夠使用 SGD 保存訓練我的神經網絡的狀態。 我知道我可以使用 save_weights 函數保存權重。 我不知道的是如何保存梯度體面的動量。 有誰知道如何做到這一點?
使用model.optimizer.get_state()
和model.optimizer.set_state()
- 請參閱https://github.com/fchollet/keras/issues/454 。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.