gpt4 book ai didi

python - 如何在 Keras 训练过程中保存动力?

转载 作者:太空宇宙 更新时间:2023-11-03 16:36:41 25 4
gpt4 key购买 nike

我希望能够保存使用 SGD 训练神经网络的状态。我知道我可以使用 save_weights 函数保存权重。我不知道的是如何保存梯度体面的动量。有谁知道该怎么做吗?

最佳答案

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

关于python - 如何在 Keras 训练过程中保存动力?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37151872/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com