未验证 提交 314cc495 编写于 作者: Z Zeng Jinle 提交者: GitHub

Revert "fix the initializer of resnet unit op (#36483)" (#36487)

This reverts commit 0452f27c.
上级 0452f27c
......@@ -187,7 +187,9 @@ class ResNetUnit(Layer):
filter_z_shape = [num_filters, filter_size, filter_size, num_channels_z]
self.filter_x = self.create_parameter(
shape=filter_x_shape, attr=filter_x_attr, default_initializer=None)
shape=filter_x_shape,
attr=filter_x_attr,
default_initializer=_get_default_param_initializer(num_channels_x))
self.scale_x = self.create_parameter(
shape=bn_param_shape,
attr=scale_x_attr,
......@@ -218,7 +220,8 @@ class ResNetUnit(Layer):
self.filter_z = self.create_parameter(
shape=filter_z_shape,
attr=filter_z_attr,
default_initializer=None)
default_initializer=_get_default_param_initializer(
num_channels_z))
self.scale_z = self.create_parameter(
shape=bn_param_shape,
attr=scale_z_attr,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册