rm to tensor

2022-04-12 19:39:50 +08:00 · 2022-04-12 19:39:50 +08:00 · 0f5fb9bd35
parent a8318353be
commit 0f5fb9bd35
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):