未验证 提交 79656c59 编写于 作者: W whs 提交者: GitHub

Fix ocr attention for python3. (#1315)

上级 14285daa
...@@ -2,6 +2,7 @@ from __future__ import absolute_import ...@@ -2,6 +2,7 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import paddle.fluid as fluid import paddle.fluid as fluid
import six
decoder_size = 128 decoder_size = 128
word_vector_dim = 128 word_vector_dim = 128
...@@ -22,7 +23,7 @@ def conv_bn_pool(input, ...@@ -22,7 +23,7 @@ def conv_bn_pool(input,
pool=True, pool=True,
use_cudnn=True): use_cudnn=True):
tmp = input tmp = input
for i in xrange(group): for i in six.moves.xrange(group):
filter_size = 3 filter_size = 3
conv_std = (2.0 / (filter_size**2 * tmp.shape[1]))**0.5 conv_std = (2.0 / (filter_size**2 * tmp.shape[1]))**0.5
conv_param = fluid.ParamAttr( conv_param = fluid.ParamAttr(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册