add options to estimate k mixture model

1 files changed, 5 insertions, 3 deletions

diff --git a/utils/training/gen_ngram.cpp b/utils/training/gen_ngram.cpp
index 918c9c8..367728a 100644
--- a/utils/training/gen_ngram.cpp
+++ b/utils/training/gen_ngram.cpp
@@ -35,12 +35,12 @@ void print_help(){
 }
 
 int main(int argc, char * argv[]){
-	int i = 1;
+    int i = 1;
     bool train_pi_gram = true;
     bool train_unigram = true;
     const char * bigram_filename = "../../data/bigram.db";
 
-    setlocale(LC_ALL,"");
+    setlocale(LC_ALL, "");
     while ( i < argc ){
 	if ( strcmp("--help", argv[i] ) == 0){
 	    print_help();
@@ -52,7 +52,9 @@ int main(int argc, char * argv[]){
             if ( ++i >= argc )
                 print_help();
             bigram_filename = argv[i];
-	}
+	}else{
+            print_help();
+        }
 	++i;
     }