From ba99237f1ab258e8f5eac0d01719e92751c9b2a3 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 14 May 2018 23:27:16 +0200
Subject: okayyyyyyy

---
 gen-prime.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'gen-prime.sh')

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 \
-- 
cgit v1.2.3-70-g09d2