From 43b09f10a4199e0dc6b5545fc06f6ee839c65c9d Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Mon, 18 Nov 2019 19:00:00 -0700 Subject: [PATCH] update --- generator/gpt2/gpt2_generator.py | 2 +- generator/gpt2/src/sample.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 234ba91..e7a5251 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -11,7 +11,7 @@ tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR) class GPT2Generator: - def __init__(self, generate_num=80, temperature=0.3, top_k=40, top_p=0.8): + def __init__(self, generate_num=80, temperature=0.2, top_k=40, top_p=0.8): self.generate_num=generate_num self.temp = temperature self.top_k = top_k diff --git a/generator/gpt2/src/sample.py b/generator/gpt2/src/sample.py index 220ff1d..1efc334 100644 --- a/generator/gpt2/src/sample.py +++ b/generator/gpt2/src/sample.py @@ -17,7 +17,7 @@ def penalize_used(logits, output): return tf.compat.v1.where( bool_tensor, - logits / 1.1, + logits / 1.15, logits)