wassname/Clover-Edition
1
0
Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-09-10 11:40:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Nick Walton
2019-11-14 17:05:17 -07:00
parent 6d82fd855e
commit 22a1c95490
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/simple/simple_generator.py
+3 -3
View File
@@ -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...")
story/story_data.yaml
+1 -1
View File
@@ -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. "
classic: "You are a knight in a deep dungeon. "