From 1db6b7a3195a3b2a2d2c8fdd759242fd57cc18fc Mon Sep 17 00:00:00 2001 From: Nick Date: Wed, 18 Sep 2019 23:02:03 -0600 Subject: [PATCH] update fix --- 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 7361068..b87b9b8 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -14,7 +14,7 @@ pos_action_starts = ["You attack", "You tell", "You use", "You go"] # the loss function is a simple categorical crossentropy between the logits and the labels def loss(labels, logits): - return tf.keras.losses.spae_categorical_crossentropy(labels, logits, from_logits=True) + return tf.keras.losses.sparse_categorical_crossentropy(labels, logits, from_logits=True) class CTRLGenerator(): diff --git a/generator/ctrl/install_ctrl.sh b/generator/ctrl/install_ctrl.sh index 3667751..42cb2df 100755 --- a/generator/ctrl/install_ctrl.sh +++ b/generator/ctrl/install_ctrl.sh @@ -20,6 +20,7 @@ cd .. curl https://sdk.cloud.google.com | bash exec -l $SHELL gcloud init +export CUDA_VISIBLE_DEVICES=0 # Download the 512-length model if specified, 256-length otherwise #if [ "$1" = "512" ]