Update GradientDescent.java

linqs · Aug 25, 2023 · 4b1ce6c · 4b1ce6c
1 parent 1d54995
commit 4b1ce6c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...ore/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java b/...ore/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java
@@ -231,7 +231,7 @@ protected void doLearn() {
                 log.trace("{}", weightedRule);
             }
 
-            if (log.isTraceEnabled() && (evaluation != null) && (iteration % trainingEvaluationComputePeriod == 0)) {
+            if (log.isTraceEnabled() && (evaluation != null) && (epoch % trainingEvaluationComputePeriod == 0)) {
                 runMAPEvaluation();
                 log.trace("MAP State Training Evaluation Metric: {}", evaluation.getNormalizedRepMetric());
 
@@ -241,7 +241,7 @@ protected void doLearn() {
                 }
             }
 
-            if (runValidation && (iteration % validationEvaluationComputePeriod == 0)) {
+            if (runValidation && (epoch % validationEvaluationComputePeriod == 0)) {
                 runValidationEvaluation();
                 log.debug("Current MAP State Validation Evaluation Metric: {}", currentValidationEvaluationMetric);