未验证 提交 0fff666f 编写于 作者: X Xin Pan 提交者: GitHub

Merge pull request #16449 from panyx0718/imperative3

fix imperative varbase creation
...@@ -403,8 +403,8 @@ class Variable(object): ...@@ -403,8 +403,8 @@ class Variable(object):
self._ivar = core.VarBase( self._ivar = core.VarBase(
name, dtype if dtype else core.VarDesc.VarType.FP32, name, dtype if dtype else core.VarDesc.VarType.FP32,
list(shape) if shape else [], list(shape) if shape else [],
_current_expected_place(), True _current_expected_place(), stop_gradient, True
if persistable else False, stop_gradient) if persistable else False)
if persistable: if persistable:
_imperative_tracer().trace_var(name, self) _imperative_tracer().trace_var(name, self)
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册