From b1aca8ba95b718aef5e09ea26eaaf1b25f7e34a5 Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 19 Sep 2019 23:02:43 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 8b585ad..73c844b 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -146,7 +146,9 @@ class CTRLGenerator(): prompt = self.control_code + prompt - print("\n\n Prompt to generate by is \n", prompt) + print("******************************") + print(" DEBUG:: Prompt to generate by is \n", prompt) + print("******************************") prompt_length = len(prompt) @@ -246,7 +248,7 @@ class CTRLGenerator(): # like which one was chosen, # what the top25 were, # here is your opportunity. - # print('chosen:', idx2word[idx]) + print('chosen:', self.idx2word[idx]) # print('top25 alternatives:', pruned_list[:25]) # assign the token for generation