Update word2vec.c

Free the dynamically allocated memory once all-thread is joined.
dav · May 10, 2021 · 2cc66fe · 2cc66fe
1 parent 9b8b580
commit 2cc66fe
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");