add run.sh

This commit is contained in:
JINCHONGJIU
2019-12-06 15:03:28 +08:00
committed by GitHub
parent 8baa4f6bf9
commit a32662a5e7

View File

@@ -0,0 +1,70 @@
#!/bin/bash
if [ "$1" = "train" ]; then
CUDA_VISIBLE_DEVICES=0 python run.py train --train-src=./en_es_data/train.es --train-tgt=./en_es_data/train.en \
--dev-src=./en_es_data/dev.es --dev-tgt=./en_es_data/dev.en --vocab=vocab.json --cuda
elif [ "$1" = "test" ]; then
mkdir -p outputs
touch outputs/test_outputs.txt
CUDA_VISIBLE_DEVICES=0 python run.py decode model.bin ./en_es_data/test.es ./en_es_data/test.en outputs/test_outputs.txt --cuda
elif [ "$1" = "train_local_q1" ]; then
python run.py train --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \
--dev-src=./en_es_data/dev_tiny.es --dev-tgt=./en_es_data/dev_tiny.en --vocab=vocab_tiny_q1.json --batch-size=2 \
--valid-niter=100 --max-epoch=101 --no-char-decoder --cuda
elif [ "$1" = "test_local_q1" ]; then
mkdir -p outputs
touch outputs/test_local_outputs.txt
python run.py decode model.bin ./en_es_data/test_tiny.es ./en_es_data/test_tiny.en outputs/test_outputs_local_q1.txt \
--no-char-decoder --cuda
elif [ "$1" = "train_local_q2" ]; then
python run.py train --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \
--dev-src=./en_es_data/dev_tiny.es --dev-tgt=./en_es_data/dev_tiny.en --vocab=vocab_tiny_q2.json --batch-size=2 \
--max-epoch=201 --valid-niter=100 --cuda
elif [ "$1" = "test_local_q2" ]; then
mkdir -p outputs
touch outputs/test_local_outputs.txt
python run.py decode model.bin ./en_es_data/test_tiny.es ./en_es_data/test_tiny.en outputs/test_outputs_local_q2.txt #!/bin/bash
if [ "$1" = "train" ]; then
CUDA_VISIBLE_DEVICES=0 python run.py train --train-src=./en_es_data/train.es --train-tgt=./en_es_data/train.en \
--dev-src=./en_es_data/dev.es --dev-tgt=./en_es_data/dev.en --vocab=vocab.json --cuda
elif [ "$1" = "test" ]; then
mkdir -p outputs
touch outputs/test_outputs.txt
CUDA_VISIBLE_DEVICES=0 python run.py decode model.bin ./en_es_data/test.es ./en_es_data/test.en outputs/test_outputs.txt --cuda
elif [ "$1" = "train_local_q1" ]; then
python run.py train --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \
--dev-src=./en_es_data/dev_tiny.es --dev-tgt=./en_es_data/dev_tiny.en --vocab=vocab_tiny_q1.json --batch-size=2 \
--valid-niter=100 --max-epoch=101 --no-char-decoder --cuda
elif [ "$1" = "test_local_q1" ]; then
mkdir -p outputs
touch outputs/test_local_outputs.txt
python run.py decode model.bin ./en_es_data/test_tiny.es ./en_es_data/test_tiny.en outputs/test_outputs_local_q1.txt \
--no-char-decoder --cuda
elif [ "$1" = "train_local_q2" ]; then
python run.py train --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \
--dev-src=./en_es_data/dev_tiny.es --dev-tgt=./en_es_data/dev_tiny.en --vocab=vocab_tiny_q2.json --batch-size=2 \
--max-epoch=201 --valid-niter=100 --cuda
elif [ "$1" = "test_local_q2" ]; then
mkdir -p outputs
touch outputs/test_local_outputs.txt
python run.py decode model.bin ./en_es_data/test_tiny.es ./en_es_data/test_tiny.en outputs/test_outputs_local_q2.txt
elif [ "$1" = "vocab" ]; then
python vocab.py --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \
--size=200 --freq-cutoff=1 vocab_tiny_q1.json
python vocab.py --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \
vocab_tiny_q2.json
python vocab.py --train-src=./en_es_data/train.es --train-tgt=./en_es_data/train.en vocab.json
else
echo "Invalid Option Selected"
fi
elif [ "$1" = "vocab" ]; then
python vocab.py --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \
--size=200 --freq-cutoff=1 vocab_tiny_q1.json
python vocab.py --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \
vocab_tiny_q2.json
python vocab.py --train-src=./en_es_data/train.es --train-tgt=./en_es_data/train.en vocab.json
else
echo "Invalid Option Selected"
fi