From c0707e56f7ef1c1d5b2121ca1a0db826c0fecf59 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Thu, 26 Sep 2019 13:38:08 -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 2fae2ea..0c9e30a 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="Horror Text: ", generate_num=100, temperature=0.3): + def __init__(self, control_code="Horror Text: ", generate_num=100, temperature=0.7): self.generate_num=generate_num model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/"