From 33d9688014833434618cfbdd0a14a633e46e83c7 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Tue, 10 Nov 2020 17:25:44 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0NumpyArrayInitializer?= =?UTF-8?q?=E4=B8=BAAssign?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ppocr/modeling/transform/tps.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/ppocr/modeling/transform/tps.py b/ppocr/modeling/transform/tps.py index f5b4f60b..50c1740e 100644 --- a/ppocr/modeling/transform/tps.py +++ b/ppocr/modeling/transform/tps.py @@ -102,13 +102,11 @@ class LocalizationNetwork(nn.Layer): name = "loc_fc2" param_attr = ParamAttr( learning_rate=loc_lr, - initializer=paddle.fluid.initializer.NumpyArrayInitializer( - np.zeros([fc_dim, F * 2])), + initializer=nn.initializer.Assign(np.zeros([fc_dim, F * 2])), name=name + "_w") bias_attr = ParamAttr( learning_rate=loc_lr, - initializer=paddle.fluid.initializer.NumpyArrayInitializer( - initial_bias), + initializer=nn.initializer.Assign(initial_bias), name=name + "_b") self.fc2 = nn.Linear( fc_dim, -- GitLab