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

swt-req
hong 4 years ago committed by GitHub
parent af70924061
commit 5a589b2f86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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):
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):
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

Loading…
Cancel
Save