From ae7ec784ff1664d46f60364426d7c6d2e016da92 Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 19 Sep 2019 11:39:00 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 5c92ec3..52810a7 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -152,7 +152,7 @@ class CTRLGenerator(): total_text_len = len(text) + self.generate_num # pad with 0s and create a mini-batch of 2 (arbitrary, for ease of code) - padded_text = text + [0] * (self.total_text_len - len(text)) + padded_text = text + [0] * (total_text_len - len(text)) tokens_generated = np.tile(padded_text, (1, 1)) result = "" for token in range(len(text) - 1, total_text_len - 1):