This commit is contained in:
Nick committed 2019-10-29 09:29:12 -06:00
1 parent c9a65db512
commit 05a6c509fb
2 files changed
+2 -2

No files matched your search

+1 -1
View File
@@ -20,7 +20,7 @@ def loss(labels, logits):
class CTRLGenerator():
def __init__(self, control_code="Writing Text:", generate_num=45, temperature=0.5, topk=40, nucleus_prob=0):
def __init__(self, control_code="Writing Text: ", generate_num=45, temperature=0.5, topk=40, nucleus_prob=0):
self.generate_num=generate_num
model_dir = "generator/ctrl/training_utils/seqlen256_v1.ckpt/"
+1 -1
View File
@@ -23,4 +23,4 @@ rooms:
haunted_hospital: ["lobby", "hallway", "parking", "roof", "pharmacy"]
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."
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. "