diff options
| -rwxr-xr-x | gen-prime.sh | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/gen-prime.sh b/gen-prime.sh index 2c2b80e..9846bcb 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 \ @@ -27,8 +27,8 @@ function gen_prime () { --primer_a $primer_a \ --primer_b $primer_b \ --recursive $recursive \ - --cuda False \ --resume True + # --cuda False \ tag="${primer}" # _${sample_length}" |