From 5d5ea2acdb95fcdf4a41e63c0e811f8c8996a8b9 Mon Sep 17 00:00:00 2001 From: Nick Date: Tue, 24 Sep 2019 15:49:48 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 83429de..f9a2550 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -324,6 +324,8 @@ class CTRLGenerator(): if self.idx2word[idx] is "\n": num_new_lines += 1 + print("Generated token is ", self.idx2word[idx]) + tokens_generated_so_far = ' '.join([self.idx2word[c] for c in tokens_generated[0][len(text):].squeeze()[:token + 2]]) tokens_generated_so_far = re.sub('(@@ )', '', string=tokens_generated_so_far) tokens_generated_so_far = re.sub('(@@ ?$)', '', string=tokens_generated_so_far)