Skip to content

GitLab

Commit 33d96880 authored by WenmuZhou's avatar WenmuZhou
Browse files

更新NumpyArrayInitializer为Assign

parent 65d3dfc7
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
ppocr/modeling/transform/tps.py
View file @ 33d96880
...@@ -102,13 +102,11 @@ class LocalizationNetwork(nn.Layer): ...@@ -102,13 +102,11 @@ class LocalizationNetwork(nn.Layer):
name = "loc_fc2" name = "loc_fc2"
param_attr = ParamAttr( param_attr = ParamAttr(
learning_rate=loc_lr, learning_rate=loc_lr,
initializer=paddle.fluid.initializer.NumpyArrayInitializer( initializer=nn.initializer.Assign(np.zeros([fc_dim, F * 2])),
np.zeros([fc_dim, F * 2])),
name=name + "_w") name=name + "_w")
bias_attr = ParamAttr( bias_attr = ParamAttr(
learning_rate=loc_lr, learning_rate=loc_lr,
initializer=paddle.fluid.initializer.NumpyArrayInitializer( initializer=nn.initializer.Assign(initial_bias),
initial_bias),
name=name + "_b") name=name + "_b")
self.fc2 = nn.Linear( self.fc2 = nn.Linear(
fc_dim, fc_dim,
......
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