提交 6b95a8a8 编写于 作者: K Kexin Zhao

fix error

上级 e09ac3df
...@@ -218,10 +218,10 @@ def decode_main(use_cuda, is_sparse): ...@@ -218,10 +218,10 @@ def decode_main(use_cuda, is_sparse):
init_recursive_seq_lens = [1] * batch_size init_recursive_seq_lens = [1] * batch_size
init_recursive_seq_lens = [init_recursive_seq_lens, init_recursive_seq_lens] init_recursive_seq_lens = [init_recursive_seq_lens, init_recursive_seq_lens]
init_ids = fluid.create_lod_tensor(init_ids_data, init_ids = fluid.create_lod_tensor(init_ids_data, init_recursive_seq_lens,
init_init_recursive_seq_lens, place) place)
init_scores = fluid.create_lod_tensor(init_scores_data, init_scores = fluid.create_lod_tensor(init_scores_data,
init_init_recursive_seq_lens, place) init_recursive_seq_lens, place)
train_data = paddle.batch( train_data = paddle.batch(
paddle.reader.shuffle( paddle.reader.shuffle(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册