提交 f70948ab authored 作者: Pascal Lamblin's avatar Pascal Lamblin 提交者: GitHub

Merge pull request #4871 from ozancaglayan/scan-fix-dict

scan/scan_op: Convert known_grads to OrderedDict
...@@ -2024,7 +2024,7 @@ class Scan(PureOp): ...@@ -2024,7 +2024,7 @@ class Scan(PureOp):
# it will be the sum of the external gradient signal and the # it will be the sum of the external gradient signal and the
# gradient obtained by propagating Y's external gradient signal # gradient obtained by propagating Y's external gradient signal
# to X. # to X.
known_grads = dict([(k.copy(), v) for (k, v) in known_grads.items()]) known_grads = OrderedDict([(k.copy(), v) for (k, v) in known_grads.items()])
grads = gradient.grad( grads = gradient.grad(
cost=None, cost=None,
...@@ -2094,7 +2094,7 @@ class Scan(PureOp): ...@@ -2094,7 +2094,7 @@ class Scan(PureOp):
dC_dXts.append(dC_dXt) dC_dXts.append(dC_dXt)
known_grads = {} known_grads = OrderedDict()
dc_dxts_idx = 0 dc_dxts_idx = 0
for i in range(len(diff_outputs)): for i in range(len(diff_outputs)):
if i < idx_nitsot_start or i >= idx_nitsot_end: if i < idx_nitsot_start or i >= idx_nitsot_end:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论