Merge pull request #5954 from WenmuZhou/whl

rm to tensor in fce head
2022-04-13 23:41:41 +08:00 · 2022-04-13 23:41:41 +08:00 · b8e7c6cef7
parent a2050ad123 0f5fb9bd35
commit b8e7c6cef7
1 changed files with 2 additions and 2 deletions
--- a/ppocr/modeling/heads/det_fce_head.py
+++ b/ppocr/modeling/heads/det_fce_head.py
@ -63,7 +63,7 @@ class FCEHead(nn.Layer):
            weight_attr=ParamAttr(
                name='cls_weights',
                initializer=Normal(
-                    mean=paddle.to_tensor(0.), std=paddle.to_tensor(0.01))),
+                    mean=0., std=0.01)),
            bias_attr=True)
        self.out_conv_reg = nn.Conv2D(
            in_channels=self.in_channels,
@ -75,7 +75,7 @@ class FCEHead(nn.Layer):
            weight_attr=ParamAttr(
                name='reg_weights',
                initializer=Normal(
-                    mean=paddle.to_tensor(0.), std=paddle.to_tensor(0.01))),
+                    mean=0., std=0.01)),
            bias_attr=True)

    def forward(self, feats, targets=None):