1 files changed, 1 insertions, 1 deletions
diff --git a/gen-prime.sh b/gen-prime.sh
index 28c9ddf..19ef340 100755
--- a/gen-prime.sh
+++ b/gen-prime.sh
@@ -12,7 +12,7 @@ function gen_prime () {
   echo "___________________________________________________"
   echo ">> generating $exp_name"
   echo ""
-  python generate.py \
+  CUDA_LAUNCH_BLOCKING=1 python generate.py \
     --exp $exp_name --dataset $exp_name \
     --frame_sizes 8 2                   \
     --n_rnn 2 --dim 1024 --q_levels 256 \