1 files changed, 10 insertions, 2 deletions
diff --git a/utils/training/gen_unigram.cpp b/utils/training/gen_unigram.cpp
index 022f159..cd938f6 100644
--- a/utils/training/gen_unigram.cpp
+++ b/utils/training/gen_unigram.cpp
@@ -31,12 +31,20 @@ int main(int argc, char * argv[]){
     
     /* gb_char binary file */
     MemoryChunk * chunk = new MemoryChunk;
-    chunk->load("gb_char.bin");
+    bool retval = chunk->load("gb_char.bin");
+    if (!retval) {
+        fprintf(stderr, "open gb_char.bin failed!\n");
+        exit(ENOENT);
+    }
     phrase_index.load(1, chunk);
     
     /* gbk_char binary file */
     chunk = new MemoryChunk;
-    chunk->load("gbk_char.bin");
+    retval = chunk->load("gbk_char.bin");
+    if (!retval) {
+        fprintf(stderr, "open gbk_char.bin failed!\n");
+        exit(ENOENT);
+    }
     phrase_index.load(2, chunk);
 
     /* Note: please increase the value when corpus size becomes larger.