diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-05-14 22:15:32 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-05-14 22:15:32 +0200 |
| commit | 39e498804bd1e799354cb8da3644465aac8d5ce9 (patch) | |
| tree | c35b8a29c31b1bfbf7a25ac3e61d75f292a6e9c3 | |
| parent | 0e7921c8fa6b496dedafb02a236ad3e19531132d (diff) | |
cuda_launch_blocking
| -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}" |