diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index a5fb09f..2377ca7 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -323,7 +323,7 @@ class CTRLGenerator(): if self.idx2word[idx] is "\n": num_new_lines += 1 - #print("Generated token is ", repr(self.idx2word[idx])) + print(self.idx2word[idx], end=" ") 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)