https://github.com/stwisdom/urnn
Raw File
Tip revision: 9f8b679c03683d0edd3f9a38d5a7cd0eef25a1c5 authored by Scott T Wisdom on 28 April 2017, 20:34:56 UTC
Update README.md
Tip revision: 9f8b679
run_memory_problem.sh
#!/bin/bash
niter=10000
batch_size=20
learning_rate=0.001

mkdir -p exp

for t in 1000 2000; do
    for model in LSTM,68,adhoc complex_RNN,128,full complex_RNN,470,adhoc ; do IFS=","; set $model
        SECONDS=0
        w_impl=$3
        echo "Running memory_problem experiment for N=$2 $1 with time_steps=$t"
        cmd="THEANO_FLAGS='device=gpu0' python2.7 -u memory_problem.py $niter ${batch_size} $2 $t 0.001 exp/memory_problem_$1_$3_$1_nhidden$2_t$t $1 categorical True CE $w_impl"
        echo $cmd
        eval $cmd
        echo "Experiment took $SECONDS seconds."
    done
done

back to top