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

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

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