From 25614b0b8613cb2bab2b15c66a7797b847749685 Mon Sep 17 00:00:00 2001 From: Nick Date: Tue, 29 Oct 2019 10:02:52 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 6bfbe5c..62d4489 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -20,7 +20,7 @@ def loss(labels, logits): class CTRLGenerator(): - def __init__(self, control_code="Writing Text: ", generate_num=45, temperature=0.5, topk=40, nucleus_prob=0): + def __init__(self, control_code="Writing Text: ", generate_num=45, temperature=0.25, topk=50, nucleus_prob=0): self.generate_num=generate_num model_dir = "generator/ctrl/training_utils/seqlen256_v1.ckpt/" @@ -219,7 +219,7 @@ class CTRLGenerator(): forbidden_tokens = ['', 'Sco@@', "&@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@", "Edit", "&@@", "2:","1:", ":", "Edit@@", "EDI@@", "EDIT@@", "edit", "TL@@", "tl@@", ";@@", '**', "http://@@", "Redd@@", "UP@@", "mom", "Up@@", "Me:", "Update", "mom@@", "Part", - "http://www.@@", "edit@@", "*@@", "\n", "Writing", "Text@@"] + "http://www.@@", "edit@@", "*@@", "\n", "Writing", "Text@@", "[[@@]", "\\@@", "
@@", " self.max_new_lines: forbidden_tokens.append("\n")