【发布时间】:2019-06-28 11:49:53
【问题描述】:
如何在 keras 中重置优化器状态?
查看优化器类我看不到这样的方法: https://github.com/keras-team/keras/blob/613aeff37a721450d94906df1a3f3cc51e2299d4/keras/optimizers.py#L60
self.updates 和 self.weights 究竟是什么?
【问题讨论】:
-
这确实不是一个简单的任务! ML 应该很难,不是吗,亲爱的 TF 团队?与此问题相关的优化器的一个流行用例是从检查点保存/加载它们。此用例的解决方法是不保存它们。为此有一个特殊的标志:
model.save(... , include_optimizer=False)!
标签: python tensorflow keras