diff --git a/src/jose.c b/src/jose.c index 6d518c2..85b7818 100755 --- a/src/jose.c +++ b/src/jose.c @@ -309,8 +309,11 @@ void LoadEmb(char *emb_file, real *emb_ptr) { } // read embedding file FILE *fp = fopen(emb_file, "r"); - fscanf(fp, "%d", &vocab_size_tmp); - fscanf(fp, "%d", &word_dim); + if (fscanf(fp, "%d", &vocab_size_tmp) != 1 || fscanf(fp, "%d", &word_dim) != 1) { + printf("Embedding file read failed\n"); + fclose(fp); + exit(1); + } if (layer1_size != word_dim) { printf("Embedding dimension incompatible with pretrained file!\n"); exit(1);