|
| 1 | +#!/bin/bash |
| 2 | + |
| 3 | +if [ "$1" = "train" ]; then |
| 4 | + CUDA_VISIBLE_DEVICES=0 python run.py train --train-src=./en_es_data/train.es --train-tgt=./en_es_data/train.en \ |
| 5 | + --dev-src=./en_es_data/dev.es --dev-tgt=./en_es_data/dev.en --vocab=vocab.json --cuda |
| 6 | +elif [ "$1" = "test" ]; then |
| 7 | + mkdir -p outputs |
| 8 | + touch outputs/test_outputs.txt |
| 9 | + 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 |
| 10 | +elif [ "$1" = "train_local_q1" ]; then |
| 11 | + python run.py train --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \ |
| 12 | + --dev-src=./en_es_data/dev_tiny.es --dev-tgt=./en_es_data/dev_tiny.en --vocab=vocab_tiny_q1.json --batch-size=2 \ |
| 13 | + --valid-niter=100 --max-epoch=101 --no-char-decoder --cuda |
| 14 | +elif [ "$1" = "test_local_q1" ]; then |
| 15 | + mkdir -p outputs |
| 16 | + touch outputs/test_local_outputs.txt |
| 17 | + python run.py decode model.bin ./en_es_data/test_tiny.es ./en_es_data/test_tiny.en outputs/test_outputs_local_q1.txt \ |
| 18 | + --no-char-decoder --cuda |
| 19 | +elif [ "$1" = "train_local_q2" ]; then |
| 20 | + python run.py train --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \ |
| 21 | + --dev-src=./en_es_data/dev_tiny.es --dev-tgt=./en_es_data/dev_tiny.en --vocab=vocab_tiny_q2.json --batch-size=2 \ |
| 22 | + --max-epoch=201 --valid-niter=100 --cuda |
| 23 | +elif [ "$1" = "test_local_q2" ]; then |
| 24 | + mkdir -p outputs |
| 25 | + touch outputs/test_local_outputs.txt |
| 26 | + 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 |
| 27 | + |
| 28 | +if [ "$1" = "train" ]; then |
| 29 | + CUDA_VISIBLE_DEVICES=0 python run.py train --train-src=./en_es_data/train.es --train-tgt=./en_es_data/train.en \ |
| 30 | + --dev-src=./en_es_data/dev.es --dev-tgt=./en_es_data/dev.en --vocab=vocab.json --cuda |
| 31 | +elif [ "$1" = "test" ]; then |
| 32 | + mkdir -p outputs |
| 33 | + touch outputs/test_outputs.txt |
| 34 | + 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 |
| 35 | +elif [ "$1" = "train_local_q1" ]; then |
| 36 | + python run.py train --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \ |
| 37 | + --dev-src=./en_es_data/dev_tiny.es --dev-tgt=./en_es_data/dev_tiny.en --vocab=vocab_tiny_q1.json --batch-size=2 \ |
| 38 | + --valid-niter=100 --max-epoch=101 --no-char-decoder --cuda |
| 39 | +elif [ "$1" = "test_local_q1" ]; then |
| 40 | + mkdir -p outputs |
| 41 | + touch outputs/test_local_outputs.txt |
| 42 | + python run.py decode model.bin ./en_es_data/test_tiny.es ./en_es_data/test_tiny.en outputs/test_outputs_local_q1.txt \ |
| 43 | + --no-char-decoder --cuda |
| 44 | +elif [ "$1" = "train_local_q2" ]; then |
| 45 | + python run.py train --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \ |
| 46 | + --dev-src=./en_es_data/dev_tiny.es --dev-tgt=./en_es_data/dev_tiny.en --vocab=vocab_tiny_q2.json --batch-size=2 \ |
| 47 | + --max-epoch=201 --valid-niter=100 --cuda |
| 48 | +elif [ "$1" = "test_local_q2" ]; then |
| 49 | + mkdir -p outputs |
| 50 | + touch outputs/test_local_outputs.txt |
| 51 | + python run.py decode model.bin ./en_es_data/test_tiny.es ./en_es_data/test_tiny.en outputs/test_outputs_local_q2.txt |
| 52 | +elif [ "$1" = "vocab" ]; then |
| 53 | + python vocab.py --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \ |
| 54 | + --size=200 --freq-cutoff=1 vocab_tiny_q1.json |
| 55 | + python vocab.py --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \ |
| 56 | + vocab_tiny_q2.json |
| 57 | + python vocab.py --train-src=./en_es_data/train.es --train-tgt=./en_es_data/train.en vocab.json |
| 58 | +else |
| 59 | + echo "Invalid Option Selected" |
| 60 | +fi |
| 61 | + |
| 62 | +elif [ "$1" = "vocab" ]; then |
| 63 | + python vocab.py --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \ |
| 64 | + --size=200 --freq-cutoff=1 vocab_tiny_q1.json |
| 65 | + python vocab.py --train-src=./en_es_data/train_tiny.es --train-tgt=./en_es_data/train_tiny.en \ |
| 66 | + vocab_tiny_q2.json |
| 67 | + python vocab.py --train-src=./en_es_data/train.es --train-tgt=./en_es_data/train.en vocab.json |
| 68 | +else |
| 69 | + echo "Invalid Option Selected" |
| 70 | +fi |
0 commit comments