decode.sh 516 Bytes
Newer Older
1 2 3 4
#! /bin/bash

gpu_num=1

5 6
data_dir=
test_subset=test
7 8 9 10 11 12 13 14

exp_name=
if [ "$#" -eq 1 ]; then
    exp_name=$1
fi

n_average=5
beam_size=5
15
max_tokens=20000
16 17 18 19 20 21 22 23 24 25 26

cmd="./run.sh
    --stage 2
    --stop_stage 2
    --gpu_num ${gpu_num}
    --exp_name ${exp_name}
    --n_average ${n_average}
    --beam_size ${beam_size}
    --max_tokens ${max_tokens}
    "

27 28 29 30 31 32 33
if [[ -n ${data_dir} ]]; then
    cmd="$cmd --data_dir ${data_dir}"
fi
if [[ -n ${test_subset} ]]; then
    cmd="$cmd --test_subset ${test_subset}"
fi

34 35
echo $cmd
eval $cmd