From 56e29c886e52d8412b1ef165b26d0b5dfe9f55ec Mon Sep 17 00:00:00 2001 From: Victor Yang <547106693@qq.com> Date: Mon, 3 Dec 2018 18:41:32 -0500 Subject: [PATCH] add seperator (#167) --- vdpwi/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vdpwi/model.py b/vdpwi/model.py index 9435b1b..42fc1f9 100644 --- a/vdpwi/model.py +++ b/vdpwi/model.py @@ -77,8 +77,8 @@ class VDPWIModel(nn.Module): def create_pad_cube(self, sent1, sent2): pad_cube = [] - sent1_lengths = [len(s.split()) for s in sent1] - sent2_lengths = [len(s.split()) for s in sent2] + sent1_lengths = [len(s.split(" ")) for s in sent1] + sent2_lengths = [len(s.split(" ")) for s in sent2] max_len1 = max(sent1_lengths) max_len2 = max(sent2_lengths)