提交 e4e69c4a authored 作者: notoraptor's avatar notoraptor

Update `Op.get_params()`: look for wrapper attributes into op, not only into props.

上级 a8d23c4e
......@@ -801,7 +801,7 @@ class Op(utils.object2, PureOp, CLinkerOp):
def get_params(self, node):
if hasattr(self, 'params_type') and isinstance(self.params_type, theano.gof.wrapper.Wrapper):
wrapper = self.params_type
if hasattr(self, '__props__') and all(field in self.__props__ for field in wrapper.fields):
if all(hasattr(self, field) for field in wrapper.fields):
wrap_dict = dict()
for i in range(wrapper.length):
field = wrapper.fields[i]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论