update fix

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():

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" ]