Update main.py

Anwar-Said · Apr 27, 2023 · 04113c1 · 04113c1
1 parent 36c7a74
commit 04113c1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/DGSD/main.py b/DGSD/main.py
@@ -21,6 +21,8 @@ def get_descriptor(self, graph, bins = 50, workers=1):
             batches = np.array_split(nodes, workers)
         emb = p.starmap(self.Generate_Embeddings, zip(batches, repeat(bins)))
         embeddings = np.sum(np.array(emb),axis = 0)
+        p.close()
+        p.join()
         return embeddings
 
     def Generate_Embeddings(self, batch, nbins):