Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "3d08d8dc4e7c25b28ccfba1631e72a77ae6478be"
Commit 0f5fb9bd authored by WenmuZhou's avatar WenmuZhou
Browse files

rm to tensor

parent a8318353
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ppocr/modeling/heads/det_fce_head.py
View file @ 0f5fb9bd
...@@ -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