提交 0f5fb9bd 编写于 作者: W WenmuZhou

rm to tensor

上级 a8318353
...@@ -63,7 +63,7 @@ class FCEHead(nn.Layer): ...@@ -63,7 +63,7 @@ class FCEHead(nn.Layer):
weight_attr=ParamAttr( weight_attr=ParamAttr(
name='cls_weights', name='cls_weights',
initializer=Normal( initializer=Normal(
mean=paddle.to_tensor(0.), std=paddle.to_tensor(0.01))), mean=0., std=0.01)),
bias_attr=True) bias_attr=True)
self.out_conv_reg = nn.Conv2D( self.out_conv_reg = nn.Conv2D(
in_channels=self.in_channels, in_channels=self.in_channels,
...@@ -75,7 +75,7 @@ class FCEHead(nn.Layer): ...@@ -75,7 +75,7 @@ class FCEHead(nn.Layer):
weight_attr=ParamAttr( weight_attr=ParamAttr(
name='reg_weights', name='reg_weights',
initializer=Normal( initializer=Normal(
mean=paddle.to_tensor(0.), std=paddle.to_tensor(0.01))), mean=0., std=0.01)),
bias_attr=True) bias_attr=True)
def forward(self, feats, targets=None): def forward(self, feats, targets=None):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册