未验证 提交 963951e8 编写于 作者: Z Zhou Wei 提交者: GitHub

fix models because the gradient clip strategy has been upgraded (#4518)

上级 190a8767
...@@ -60,20 +60,16 @@ def conv_bn_pool(input, ...@@ -60,20 +60,16 @@ def conv_bn_pool(input,
def ocr_convs(input, def ocr_convs(input,
regularizer=None, regularizer=None,
gradient_clip=None,
is_test=False, is_test=False,
use_cudnn=False): use_cudnn=False):
b = fluid.ParamAttr( b = fluid.ParamAttr(
regularizer=regularizer, regularizer=regularizer,
gradient_clip=gradient_clip,
initializer=fluid.initializer.Normal(0.0, 0.0)) initializer=fluid.initializer.Normal(0.0, 0.0))
w0 = fluid.ParamAttr( w0 = fluid.ParamAttr(
regularizer=regularizer, regularizer=regularizer,
gradient_clip=gradient_clip,
initializer=fluid.initializer.Normal(0.0, 0.0005)) initializer=fluid.initializer.Normal(0.0, 0.0005))
w1 = fluid.ParamAttr( w1 = fluid.ParamAttr(
regularizer=regularizer, regularizer=regularizer,
gradient_clip=gradient_clip,
initializer=fluid.initializer.Normal(0.0, 0.01)) initializer=fluid.initializer.Normal(0.0, 0.01))
tmp = input tmp = input
tmp = conv_bn_pool( tmp = conv_bn_pool(
...@@ -119,7 +115,6 @@ def encoder_net(images, ...@@ -119,7 +115,6 @@ def encoder_net(images,
conv_features = ocr_convs( conv_features = ocr_convs(
images, images,
regularizer=regularizer, regularizer=regularizer,
gradient_clip=gradient_clip,
is_test=is_test, is_test=is_test,
use_cudnn=use_cudnn) use_cudnn=use_cudnn)
sliced_feature = fluid.layers.im2sequence( sliced_feature = fluid.layers.im2sequence(
...@@ -163,11 +158,9 @@ def encoder_net(images, ...@@ -163,11 +158,9 @@ def encoder_net(images,
w_attr = fluid.ParamAttr( w_attr = fluid.ParamAttr(
regularizer=regularizer, regularizer=regularizer,
gradient_clip=gradient_clip,
initializer=fluid.initializer.Normal(0.0, 0.02)) initializer=fluid.initializer.Normal(0.0, 0.02))
b_attr = fluid.ParamAttr( b_attr = fluid.ParamAttr(
regularizer=regularizer, regularizer=regularizer,
gradient_clip=gradient_clip,
initializer=fluid.initializer.Normal(0.0, 0.0)) initializer=fluid.initializer.Normal(0.0, 0.0))
fc_out = fluid.layers.fc(input=[gru_forward, gru_backward], fc_out = fluid.layers.fc(input=[gru_forward, gru_backward],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册