提交 25ca2ca0 编写于 作者: T Tao Luo

change init_idx to INT32 in transformer_test

test=develop
上级 e5e7e9b8
......@@ -147,7 +147,7 @@ void PrepareInputs(std::vector<PaddleTensor> *input_slots, DataRecord *data,
init_idx.name = "init_idx";
init_idx.shape.assign({batch_size});
init_idx.dtype = PaddleDType::INT64;
init_idx.dtype = PaddleDType::INT32;
TensorAssignData<int64_t>(&init_idx, one_batch.init_idx);
trg_src_attn_bias.name = "trg_src_attn_bias";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册