Fix for ParallelTokenization==false

deeplearning4j · Jan 21, 2019 · 0687c79 · 0687c79
1 parent 7d08543
commit 0687c79
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...ng4j-nlp/src/main/java/org/deeplearning4j/models/word2vec/wordstore/VocabConstructor.java b/...ng4j-nlp/src/main/java/org/deeplearning4j/models/word2vec/wordstore/VocabConstructor.java
@@ -635,6 +635,7 @@ public void run() {
             } finally {
                 finalCounter.incrementAndGet();
                 synchronized (this) {
+                    done = true;
                     this.notifyAll();
                 }
             }