From 22a1c95490503f9cb1ed8c93a8744fef5b74cfeb Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Thu, 14 Nov 2019 17:05:17 -0700 Subject: [PATCH] update --- generator/simple/simple_generator.py | 6 +++--- story/story_data.yaml | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/generator/simple/simple_generator.py b/generator/simple/simple_generator.py index 2c90172..0a4d78c 100644 --- a/generator/simple/simple_generator.py +++ b/generator/simple/simple_generator.py @@ -18,14 +18,14 @@ tf.logging.set_verbosity(tf.logging.ERROR) class SimpleGenerator: - def __init__(self, generate_num=28, temperature=0.8, top_k=40, top_p=0.0): + def __init__(self, generate_num=40, temperature=0.9, top_k=40, top_p=0.0): self.generate_num=generate_num self.temp = temperature self.top_k = top_k self.top_p = top_p - self.model_name = "1558M" - self.model_dir = "generator/simple/models" + self.model_name = "run1" + self.model_dir = "generator/simple/checkpoint" self.checkpoint_path = os.path.join(self.model_dir, self.model_name) if not os.path.isdir(os.path.join(self.model_dir, self.model_name)): print(f"Downloading {self.model_name} model...") diff --git a/story/story_data.yaml b/story/story_data.yaml index 584375b..868d11c 100644 --- a/story/story_data.yaml +++ b/story/story_data.yaml @@ -31,4 +31,4 @@ contexts: apocalypse: "Long ago the bombs fell and the world ended. You are one of the few who is still alive. You are trying to survive by scavenging among the ruins of what is left behind. " - classic: "You enter a dungeon with your trusty sword and shield. You are searching for the evil necromancer who killed your family. You've heard that he resides at the bottom of the dungeon, guarded by legions of the undead. " \ No newline at end of file + classic: "You are a knight in a deep dungeon. " \ No newline at end of file