From 8ad134dbd1835abafda9f7ab859b466063c1cd72 Mon Sep 17 00:00:00 2001 From: Nick Date: Sat, 2 Nov 2019 14:24:37 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 96c5752..c479085 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -27,7 +27,7 @@ class CTRLGenerator(): self.generate_num=generate_num model_dir = "generator/ctrl/model/aidungeon2model/" - checkpoint_path = "generator/ctrl/model/aidungeon2model/checkpoint" + checkpoint_path = "generator/ctrl/model/aidungeon2model/model.ckpt-417400.data-00000-of-00002" self.control_code = control_code vocab_file = 'generator/ctrl/model/vocab' code_file = 'generator/ctrl/model/codes'