diff options
author | Rico Sennrich <rico.sennrich@gmx.ch> | 2015-02-12 14:06:41 +0300 |
---|---|---|
committer | Rico Sennrich <rico.sennrich@gmx.ch> | 2015-02-12 14:06:41 +0300 |
commit | e6fb89f3492a5f1a0fccd0cc17a50fd8be53012c (patch) | |
tree | df11bc1ef8ff23b157d46dbfd29fbf2b7a7ea7dd | |
parent | 02a3683b9f8ec536b5074a53cc2e0ec04dd3b7eb (diff) |
bugfix: don't overwrite input/output vocabulary sizes if vocab size is 0.
-rw-r--r-- | src/trainNeuralNetwork.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/trainNeuralNetwork.cpp b/src/trainNeuralNetwork.cpp index 21a70d6..c8fcb68 100644 --- a/src/trainNeuralNetwork.cpp +++ b/src/trainNeuralNetwork.cpp @@ -141,7 +141,7 @@ int main(int argc, char** argv) myParam.vocab_size = vocab_size.getValue(); myParam.input_vocab_size = input_vocab_size.getValue(); myParam.output_vocab_size = output_vocab_size.getValue(); - if (vocab_size.getValue() >= 0) { + if (vocab_size.getValue() > 0) { myParam.input_vocab_size = myParam.output_vocab_size = vocab_size.getValue(); } myParam.num_hidden = num_hidden.getValue(); @@ -324,7 +324,7 @@ int main(int argc, char** argv) for (data_size_t i=vector_size-1; i>0; i--) { if (i %500000 == 0) { - cerr<<"Shuffled "<<training_data_size-1<<" instances..."; + cerr<<"."; } data_size_t j = boost::random::uniform_int_distribution<data_size_t>(0, i-1)(rng); for (int k=0;k<myParam.ngram_size;k++) { |