@@ -10,11 +10,12 @@ if [ ! -n "$1" ] ;then
10
10
echo " you have not input a project name!"
11
11
else
12
12
echo " The project name will be set to $1 "
13
- fi
13
+
14
14
if [ ! -n " $2 " ] ; then
15
15
dimension=300
16
16
elif [[ ! $2 =~ ^[0-9]* $ ]]; then
17
17
echo " Please enter an integer [50|100|200|300] to the second parameter to set the dimension of Glove Embeddings;"
18
+ dimension=300
18
19
elif [ $2 -le 50 ]; then
19
20
dimension=50
20
21
echo " The dimension of GloVe embeddings is set to $dimension "
@@ -32,7 +33,7 @@ if [ ! -n "$3" ] ;then
32
33
num_units=512
33
34
elif [[ ! $3 =~ ^[0-9]* $ ]]; then
34
35
echo " Please enter an integer [ >=512 recommended ] to the third parameter to set the number of units of LSTM cells"
35
-
36
+ num_units=512
36
37
else
37
38
num_units=$3
38
39
echo " The number of units of LSTM cells is set to $num_units "
@@ -41,7 +42,7 @@ if [ ! -n "$4" ] ;then
41
42
training_steps=60000
42
43
elif [[ ! $4 =~ ^[0-9]* $ ]]; then
43
44
echo " Please enter an integer [ >=60000 recommended ] to the fourth parameter to set the number of training steps for Learner"
44
-
45
+ training_steps=60000
45
46
else
46
47
training_steps=$4
47
48
echo " The number of training steps for Learner is set to $training_steps "
@@ -50,7 +51,7 @@ if [ ! -n "$5" ] ;then
50
51
examples_per_template=600
51
52
elif [[ ! $5 =~ ^[0-9]* $ ]]; then
52
53
echo " Please enter an integer [ >=600 recommended ] to the fifth parameter to set the number of examples per template"
53
-
54
+ examples_per_template=600
54
55
else
55
56
examples_per_template=$5
56
57
echo " The number of examples per template is set to $examples_per_template "
137
138
python -m nmt.nmt --src=en --tgt=sparql --embed_prefix=../data/$1 /embed --vocab_prefix=../data/$1 /vocab --dev_prefix=../data/$1 /dev --test_prefix=../data/$1 /test --train_prefix=../data/$1 /train --out_dir=../data/$1 " _" $dimension " d_model" --num_train_steps=$training_steps --steps_per_stats=100 --num_layers=2 --num_units=$num_units --dropout=0.2 --metrics=bleu,accuracy
138
139
cd ..
139
140
140
-
141
+ fi
0 commit comments