未验证 提交 a40121e4 编写于 作者: Z Zhen Wang 提交者: GitHub

fix the initialization process error. test=develop (#17213)

上级 a914d9b1
...@@ -383,7 +383,7 @@ class QuantizationTransformPass(object): ...@@ -383,7 +383,7 @@ class QuantizationTransformPass(object):
data_type = 'float64' if var_node.dtype( data_type = 'float64' if var_node.dtype(
) == core.VarDesc.VarType.FP64 else 'float32' ) == core.VarDesc.VarType.FP64 else 'float32'
_init_var_node( _init_var_node(
scale_in_node, state_in_node,
np.ones( np.ones(
[1], dtype=data_type), [1], dtype=data_type),
self._scope, self._scope,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册