Skip to content

GitLab

Unverified Commit b8e7c6ce authored by zhoujun's avatar zhoujun Committed by GitHub
Browse files

Merge pull request #5954 from WenmuZhou/whl 

rm to tensor in  fce head
parents a2050ad1 0f5fb9bd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ppocr/modeling/heads/det_fce_head.py
View file @ b8e7c6ce
...@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment