提交 1838beb3 authored 作者: Yann N. Dauphin's avatar Yann N. Dauphin 提交者: GitHub

Fix crash in Lop

Fix Lop to comply with new requirement that theano.grad's known_grad argument be an OrderedDict.
上级 aecbbb99
...@@ -345,7 +345,7 @@ def Lop(f, wrt, eval_points, consider_constant=None, ...@@ -345,7 +345,7 @@ def Lop(f, wrt, eval_points, consider_constant=None,
wrt = [wrt] wrt = [wrt]
assert len(f) == len(grads) assert len(f) == len(grads)
known = dict(izip(f, grads)) known = OrderedDict(izip(f, grads))
ret = grad(cost=None, known_grads=known, ret = grad(cost=None, known_grads=known,
consider_constant=consider_constant, wrt=wrt, consider_constant=consider_constant, wrt=wrt,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论