Call next batch in runtime inference.

linqs · Nov 2, 2023 · e9c5051 · e9c5051
1 parent 1cdb8be
commit e9c5051
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/psl-java/src/main/java/org/linqs/psl/runtime/Runtime.java b/psl-java/src/main/java/org/linqs/psl/runtime/Runtime.java
@@ -448,6 +448,7 @@ protected void runInferenceInternal(RuntimeConfig config, Model model, RuntimeRe
 
             DeepPredicate.evalAllDeepPredicates();
 
+            DeepPredicate.nextBatchAllDeepPredicates();
             runInference = !DeepPredicate.isEpochCompleteAllDeepPredicates();
         }
         DeepPredicate.epochEndAllDeepPredicates();