Fix lmplz complaint about vocab_pad

jiejiang · Feb 11, 2014 · d983a14 · d983a14
1 parent 70d48ae
commit d983a14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lm/builder/lmplz_main.cc b/lm/builder/lmplz_main.cc
@@ -96,7 +96,7 @@ int main(int argc, char *argv[]) {
     }
 #endif
 
-    if (vm.count("vocab_pad") && !pipeline.initial_probs.interpolate_unigrams) {
+    if (pipeline.vocab_size_for_unk && !pipeline.initial_probs.interpolate_unigrams) {
       std::cerr << "--vocab_pad requires --interpolate_unigrams" << std::endl;
       return 1;
     }