Merge pull request #30 from sngpt08/patch-1

Memory leaks detected Issue #26
dav · May 10, 2021 · fb15097 · fb15097
2 parents 9b8b580 + 2cc66fe
commit fb15097
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/word2vec.c b/src/word2vec.c
@@ -600,6 +600,7 @@ void TrainModel() {
   pthread_t *pt = (pthread_t *)malloc(num_threads * sizeof(pthread_t));
   for (a = 0; a < num_threads; a++) pthread_create(&pt[a], NULL, TrainModelThread, (void *)a);
   for (a = 0; a < num_threads; a++) pthread_join(pt[a], NULL);
+  free(pt);
 #endif
 
   fo = fopen(output_file, "wb");