From 7e0072ab0bf8e527e95c8f27e69b7c80fdef082f Mon Sep 17 00:00:00 2001 From: wopeizl Date: Thu, 27 Jun 2019 15:32:44 +0800 Subject: [PATCH] set the feed data type to int64 since the windows would give int32 by default (#2575) test=develop --- .../deep_attention_matching/reader.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/PaddleNLP/dialogue_model_toolkit/deep_attention_matching/reader.py b/PaddleNLP/dialogue_model_toolkit/deep_attention_matching/reader.py index fd46fc32..6a2653c2 100755 --- a/PaddleNLP/dialogue_model_toolkit/deep_attention_matching/reader.py +++ b/PaddleNLP/dialogue_model_toolkit/deep_attention_matching/reader.py @@ -206,11 +206,12 @@ def make_one_batch_input(data_batches, index): feeding dictionary """ - turns = np.array(data_batches["turns"][index]) - tt_turns_len = np.array(data_batches["tt_turns_len"][index]) - every_turn_len = np.array(data_batches["every_turn_len"][index]) - response = np.array(data_batches["response"][index]) - response_len = np.array(data_batches["response_len"][index]) + turns = np.array(data_batches["turns"][index]).astype('int64') + tt_turns_len = np.array(data_batches["tt_turns_len"][index]).astype('int64') + every_turn_len = np.array(data_batches["every_turn_len"][index]).astype( + 'int64') + response = np.array(data_batches["response"][index]).astype('int64') + response_len = np.array(data_batches["response_len"][index]).astype('int64') batch_size = turns.shape[0] max_turn_num = turns.shape[1] -- GitLab