This commit is contained in:
Nick
2019-09-20 17:30:35 -06:00
parent 9d763e2afb
commit 0fe73ab7f6
2 changed files with 5 additions and 2 deletions
+4 -1
View File
@@ -57,6 +57,9 @@ def play_constrained():
result = None
while(result == None):
action_choice = input("Which action do you choose? ")
if action_choice is "print story":
print(story_manager.story)
continue
print("\n")
result, possible_actions = story_manager.act(action_choice)
@@ -84,7 +87,7 @@ def play_cached():
if __name__ == '__main__':
play_constrained()
play_unconstrained()
+1 -1
View File
@@ -22,7 +22,7 @@ def loss(labels, logits):
class CTRLGenerator():
def __init__(self, control_code="Horror Text: ", generate_num=64, temperature=0.5):
def __init__(self, control_code="Horror Text: ", generate_num=64, temperature=0.75):
self.generate_num=generate_num
model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/"