【发布时间】:2017-07-04 08:24:06
【问题描述】:
这是来自SGD 的get_updates 代码的一部分,来自keras(source)
moments = [K.zeros(shape) for shape in shapes]
self.weights = [self.iterations] + moments
for p, g, m in zip(params, grads, moments):
v = self.momentum * m - lr * g # velocity
self.updates.append(K.update(m, v))
观察:
因为moments 变量是一个零张量列表。 for loop 中的每个m 都是一个零张量,形状为p。那么self.momentum * m,在循环的第一行,只是一个标量乘以零张量,结果是零张量。
问题
我在这里缺少什么? 谢谢!
【问题讨论】:
标签: machine-learning neural-network deep-learning keras keras-2