提交 7e0072ab 编写于 作者: W wopeizl 提交者: Yibing Liu

set the feed data type to int64 since the windows would give int32 by default (#2575)

test=develop
上级 d5f74052
...@@ -206,11 +206,12 @@ def make_one_batch_input(data_batches, index): ...@@ -206,11 +206,12 @@ def make_one_batch_input(data_batches, index):
feeding dictionary feeding dictionary
""" """
turns = np.array(data_batches["turns"][index]) turns = np.array(data_batches["turns"][index]).astype('int64')
tt_turns_len = np.array(data_batches["tt_turns_len"][index]) tt_turns_len = np.array(data_batches["tt_turns_len"][index]).astype('int64')
every_turn_len = np.array(data_batches["every_turn_len"][index]) every_turn_len = np.array(data_batches["every_turn_len"][index]).astype(
response = np.array(data_batches["response"][index]) 'int64')
response_len = np.array(data_batches["response_len"][index]) 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] batch_size = turns.shape[0]
max_turn_num = turns.shape[1] max_turn_num = turns.shape[1]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册