diff options
Diffstat (limited to 'gen-prime.sh')
| -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 b4e5614..2fb638c 100755 --- a/gen-prime.sh +++ b/gen-prime.sh @@ -12,7 +12,7 @@ function gen_prime () { echo "___________________________________________________" echo ">> generating $exp_name" echo "" - CUDA_LAUNCH_BLOCKING=1 python generate.py \ + 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}" |