This commit is contained in:
Nick
2019-10-31 09:26:03 -06:00
parent 7c06efed24
commit c8a194589e
-2
View File
@@ -311,8 +311,6 @@ class CTRLGenerator():
if debug_print:
print(repr(self.idx2word[idx]), end="_")
import pdb
pdb.set_trace()
tokens_generated_so_far = ' '.join([self.idx2word[c] for c in tokens_generated[0][len(text):token+2]])
tokens_generated_so_far = re.sub('(@@ )', '', string=tokens_generated_so_far)
tokens_generated_so_far = re.sub('(@@ ?$)', '', string=tokens_generated_so_far)