未验证 提交 b3cbc5be 编写于 作者: J Jiabin Yang 提交者: GitHub

test=develop, fix test_imperative_transformer and ocr (#18127)

* test=develop, fix test_imperative_transformer and ocr

* test=develop, remove ocr recovery part
上级 ff4279e3
...@@ -177,7 +177,6 @@ set_tests_properties(test_imperative_se_resnext PROPERTIES LABELS "RUN_TYPE=EXCL ...@@ -177,7 +177,6 @@ set_tests_properties(test_imperative_se_resnext PROPERTIES LABELS "RUN_TYPE=EXCL
py_test_modules(test_install_check MODULES test_install_check ENVS py_test_modules(test_install_check MODULES test_install_check ENVS
FLAGS_cudnn_deterministic=1 SERIAL) FLAGS_cudnn_deterministic=1 SERIAL)
set_tests_properties(test_install_check PROPERTIES LABELS "RUN_TYPE=DIST") set_tests_properties(test_install_check PROPERTIES LABELS "RUN_TYPE=DIST")
if(WITH_DISTRIBUTE) if(WITH_DISTRIBUTE)
py_test_modules(test_dist_train MODULES test_dist_train) py_test_modules(test_dist_train MODULES test_dist_train)
set_tests_properties(test_listen_and_serv_op PROPERTIES TIMEOUT 20) set_tests_properties(test_listen_and_serv_op PROPERTIES TIMEOUT 20)
......
...@@ -313,22 +313,16 @@ sync = False ...@@ -313,22 +313,16 @@ sync = False
batch_num = 5 batch_num = 5
np.random.seed = 90 np.random.seed = 90
src_word_np = np.random.randint( src_word_np = np.arange(1, TrainTaskConfig.batch_size * seq_len + 1).reshape(
1, [TrainTaskConfig.batch_size, seq_len, 1]).astype('int64')
ModelHyperParams.src_vocab_size - 1,
size=(TrainTaskConfig.batch_size, seq_len, 1),
dtype='int64')
src_pos_np = np.random.randint( src_pos_np = np.random.randint(
1, seq_len, size=(TrainTaskConfig.batch_size, seq_len, 1), dtype='int64') 1, seq_len, size=(TrainTaskConfig.batch_size, seq_len, 1), dtype='int64')
src_slf_attn_bias_np = np.random.randn(TrainTaskConfig.batch_size, src_slf_attn_bias_np = np.random.randn(TrainTaskConfig.batch_size,
ModelHyperParams.n_head, seq_len, ModelHyperParams.n_head, seq_len,
seq_len).astype('float32') seq_len).astype('float32')
trg_word_np = np.random.randint( trg_word_np = np.arange(1, TrainTaskConfig.batch_size * seq_len + 1).reshape(
1, [TrainTaskConfig.batch_size, seq_len, 1]).astype('int64')
ModelHyperParams.src_vocab_size - 1,
size=(TrainTaskConfig.batch_size, seq_len, 1),
dtype='int64')
trg_pos_np = np.random.randint( trg_pos_np = np.random.randint(
1, seq_len, size=(TrainTaskConfig.batch_size, seq_len, 1), dtype='int64') 1, seq_len, size=(TrainTaskConfig.batch_size, seq_len, 1), dtype='int64')
trg_slf_attn_bias_np = np.random.randn(TrainTaskConfig.batch_size, trg_slf_attn_bias_np = np.random.randn(TrainTaskConfig.batch_size,
......
...@@ -181,22 +181,16 @@ sync = False ...@@ -181,22 +181,16 @@ sync = False
batch_num = 5 batch_num = 5
np.random.seed = 90 np.random.seed = 90
src_word_np = np.random.randint( src_word_np = np.arange(1, TrainTaskConfig.batch_size * seq_len + 1).reshape(
1, [TrainTaskConfig.batch_size, seq_len, 1]).astype('int64')
ModelHyperParams.src_vocab_size - 1,
size=(TrainTaskConfig.batch_size, seq_len, 1),
dtype='int64')
src_pos_np = np.random.randint( src_pos_np = np.random.randint(
1, seq_len, size=(TrainTaskConfig.batch_size, seq_len, 1), dtype='int64') 1, seq_len, size=(TrainTaskConfig.batch_size, seq_len, 1), dtype='int64')
src_slf_attn_bias_np = np.random.randn(TrainTaskConfig.batch_size, src_slf_attn_bias_np = np.random.randn(TrainTaskConfig.batch_size,
ModelHyperParams.n_head, seq_len, ModelHyperParams.n_head, seq_len,
seq_len).astype('float32') seq_len).astype('float32')
trg_word_np = np.random.randint( trg_word_np = np.arange(1, TrainTaskConfig.batch_size * seq_len + 1).reshape(
1, [TrainTaskConfig.batch_size, seq_len, 1]).astype('int64')
ModelHyperParams.src_vocab_size - 1,
size=(TrainTaskConfig.batch_size, seq_len, 1),
dtype='int64')
trg_pos_np = np.random.randint( trg_pos_np = np.random.randint(
1, seq_len, size=(TrainTaskConfig.batch_size, seq_len, 1), dtype='int64') 1, seq_len, size=(TrainTaskConfig.batch_size, seq_len, 1), dtype='int64')
trg_slf_attn_bias_np = np.random.randn(TrainTaskConfig.batch_size, trg_slf_attn_bias_np = np.random.randn(TrainTaskConfig.batch_size,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册