way=$1
shot=$2
i=$3
id=$4

args="../run_train.py \
		--data.dataset=googlespeech --data.way=$way --data.shot=$shot \
		--data.query=5 --data.test_way=$way --data.test_shot=$shot --data.test_query=15 \
		--data.train_episodes=200 --data.test_episodes=100 \
		--model.model_name=protonet_conv --model.x_dim=1,51,40 --model.hid_dim=64 --model.z_dim=64 \
		--train.epochs=200 --train.optim_method=Adam --train.learning_rate=0.001 --train.decay_every=20 \
		--train.weight_decay=0.0 --train.patience=200  \
		--log.exp_dir=./$id \
		--data.cuda \
        --model.encoding=TCResNet8Dilated"

if [ "$i" -eq "1" ]; then
    args+=" --speech.include_background"

elif [ "$i" -eq "2" ]; then
    args+=" --speech.include_silence"

elif [ "$i" -eq "3" ]; then
    args+=" --speech.include_unknown"

elif [ "$i" -eq "4" ]; then
    args+=" --speech.include_background"
    args+=" --speech.include_silence"

elif [ "$i" -eq "5" ]; then
    args+=" --speech.include_background"
    args+=" --speech.include_unknown"

elif [ "$i" -eq "6" ]; then
    args+=" --speech.include_unknown"
    args+=" --speech.include_silence"

elif [ "$i" -eq "7" ]; then
    args+=" --speech.include_background"
    args+=" --speech.include_unknown"
    args+=" --speech.include_silence"
fi

python $args
