未验证 提交 5a589b2f 编写于 作者: H hong 提交者: GitHub

reduce imperative ocr attention config; test=develop (#28079)

上级 af709240
......@@ -29,11 +29,11 @@ class Config(object):
config for training
'''
# encoder rnn hidden_size
encoder_size = 64
encoder_size = 16
# decoder size for decoder stage
decoder_size = 64
decoder_size = 16
# size for word embedding
word_vector_dim = 64
word_vector_dim = 16
# max length for label padding
max_length = 5
# optimizer setting
......@@ -41,9 +41,9 @@ class Config(object):
learning_rate_decay = None
# batch size to train
batch_size = 16
batch_size = 8
# class number to classify
num_classes = 481
num_classes = 64
use_gpu = False
# special label for start and end
......@@ -376,7 +376,7 @@ class TestDygraphOCRAttention(unittest.TestCase):
seed = 90
epoch_num = 1
if core.is_compiled_with_cuda():
batch_num = 10
batch_num = 6
else:
batch_num = 4
np.random.seed = seed
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册