added telemetry

This commit is contained in:
Nick Walton
2019-11-19 14:40:58 -07:00
parent 8a200b62b2
commit 2ce06037f8
3 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -10,7 +10,7 @@ import numpy as np
class GPT2Generator:
def __init__(self, generate_num=60, temperature=0.3, top_k=40, top_p=0.9):
def __init__(self, generate_num=60, temperature=0.4, top_k=40, top_p=0.9):
self.generate_num=generate_num
self.temp = temperature
self.top_k = top_k
+1 -1
View File
@@ -17,7 +17,7 @@ def penalize_used(logits, output):
return tf.compat.v1.where(
bool_tensor,
logits / 1.2,
logits * 0.85,
logits)
+1 -1
View File
@@ -34,7 +34,7 @@ def play_aidungeon_2():
save_story = input("Help improve AIDungeon by enabling story saving? (Y/n) ")
if save_story.lower() in ["no", "No", "n"]:
upload_story = True
upload_story = False
else:
upload_story = True