簡體   English   中英

如何在 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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM