From 641722fb276f0c70b837dc8cae7ccc57c7ddbb31 Mon Sep 17 00:00:00 2001 From: Nick Date: Wed, 18 Sep 2019 23:10:42 -0600 Subject: [PATCH] fixed bpe issue --- generator/ctrl/ctrl_generator.py | 2 +- generator/ctrl/install_ctrl.sh | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 30a0c30..4bad39f 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -8,7 +8,7 @@ from collections import Counter from tensorflow.python import debug as tf_debug from tensorflow.python.ops import math_ops from tensorflow.python.ops import embedding_ops -import generator.ctrl.model.fastBPE as fastBPE +import fastBPE pos_action_starts = ["You attack", "You tell", "You use", "You go"] diff --git a/generator/ctrl/install_ctrl.sh b/generator/ctrl/install_ctrl.sh index 42cb2df..99c5d2d 100755 --- a/generator/ctrl/install_ctrl.sh +++ b/generator/ctrl/install_ctrl.sh @@ -21,6 +21,7 @@ curl https://sdk.cloud.google.com | bash exec -l $SHELL gcloud init export CUDA_VISIBLE_DEVICES=0 +cd .. # Download the 512-length model if specified, 256-length otherwise #if [ "$1" = "512" ]